Named Entity Recognition (Model) for English language

This model is a fine-tuned version of google/electra-base-discriminator on tner/ontonotes5 dataset. It achieves the following results on the evaluation set:

• Loss: 0.1080
• Precision: 0.8564
• Recall: 0.8792
• F1: 0.8676
• Accuracy: 0.9741

Categories

Named Entity Types:

• CARDINAL : Numbers that do not refer to specific dates, percentages, or quantities. Examples: "three," "15," "one million."

• DATE : Any expression indicating a specific or relative date, including days, months, years, and more complex expressions. Examples: "January 1, 2020," "yesterday," "next Tuesday."

• PERSON : Names of people or fictional characters, often including full names, first names, last names, and titles. Examples: "Alice Johnson," "Dr. Smith," "Sherlock Holmes."

• NORP : Nationalities, religious groups, or political groups. Examples: "American," "Buddhist," "Democrats."

• GPE : Countries, cities, states, or any political or geographical area with defined borders. Examples: "France," "New York," "Bavaria."

• LAW : Named documents, treaties, or laws. Examples: "First Amendment," "The Treaty of Paris," "Consumer Protection Act."

• PERCENT : Percentages, represented by numbers followed by a percentage symbol or phrase. Examples: "50%," "twenty percent."

• ORDINAL : Ordinal numbers that indicate an order or rank in a sequence. Examples: "first," "3rd," "twentieth."

• MONEY : Monetary values, including units of currency. Examples: "$100," "€50 million," "ten dollars."

• WORK_OF_ART : Titles of creative works, such as books, paintings, songs, and movies. Examples: "Mona Lisa," "Inception," "The Great Gatsby."

• FAC : Buildings, airports, highways, bridges, and other physical structures or infrastructure. Examples: "Eiffel Tower," "JFK Airport," "Golden Gate Bridge."

• TIME : Times of day and exact times. Examples: "2:30 PM," "midnight," "dawn."

• QUANTITY : Measurable amounts that include units, but not money or time. Examples: "50 kilograms," "3 miles," "two liters."

• PRODUCT : Products, including vehicles, devices, and any manufactured items. Examples: "iPhone," "Boeing 747," "PlayStation."

• LANGUAGE : Names of languages. Examples: "English," "Mandarin," "Spanish."

• ORG : Names of companies, agencies, institutions, and other formal organizations. Examples: "Google," "United Nations," "Harvard University."

• LOC : Non-political geographic locations, such as mountains, bodies of water, and continents. Examples: "Mount Everest," "the Amazon River," "Europe."

• EVENT : Named historical or recurring events. Examples: "World War II," "Olympics," "Thanksgiving.

More information needed

Intended uses & limitations

Since OntoNotes includes detailed named entity annotations, a model fine-tuned on it can effectively recognize entities like people, locations, organizations, and some specialized categories.

OntoNotes primarily includes data from newswire, broadcast news, conversational telephone speech, and web data. Thus, models fine-tuned on OntoNotes may struggle with informal text like social media, domain-specific jargon, or highly technical language

Usage Example

from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification

# Specify the model checkpoint
model_checkpoint = "ShakhzoDavronov/electra-ner-token-classification"

# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
model = AutoModelForTokenClassification.from_pretrained(model_checkpoint)

# Initialize the pipeline with the model and tokenizer
nlp = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="average")

# Define a sample text for Named Entity Recognition
sample_text = '''Amazon announced its plans to open a new headquarters in Virginia, 
                 aiming to create over 25,000 jobs in the area by 2030.'''

# Run the pipeline and print results
outputs = nlp(sample_text)
for ent in outputs:
    print(ent)

Output:

{'entity_group': 'ORG', 'score': 0.9978817, 'word': 'amazon', 'start': 0, 'end': 6}
{'entity_group': 'GPE', 'score': 0.996828, 'word': 'virginia', 'start': 57, 'end': 65}
{'entity_group': 'CARDINAL', 'score': 0.638206, 'word': 'over 25, 000', 'start': 100, 'end': 111}
{'entity_group': 'DATE', 'score': 0.9956894, 'word': '2030', 'start': 132, 'end': 136}

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 2e-05
• train_batch_size: 8
• eval_batch_size: 8
• seed: 42
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• num_epochs: 3

Training results

Framework versions

• Transformers 4.44.2
• Pytorch 2.5.0+cu121
• Datasets 3.1.0
• Tokenizers 0.19.1

Contacts

If you have any questions or need more information, please contact me. LinkedIn:Shakhzod Davronov