Edit model card

bert-base-NER

Model description

bert-base-NER is a fine-tuned BERT model that is ready to use for Named Entity Recognition and achieves F1 0.61 for the NER task. It has been trained to recognize two types of entities: instrument and satellite.

Specifically, this model is a bert-base-cased model that was fine-tuned on Satellite-Instrument-NER dataset.

Intended uses & limitations

How to use

You can use this model with Transformers pipeline for NER.

from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline
tokenizer = AutoTokenizer.from_pretrained("NahedAbdelgaber/ner_base_model")
model = AutoModelForTokenClassification.from_pretrained("NahedAbdelgaber/ner_base_model")
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "Centroid Moment Tensor Global Navigation Satellite System GNSS"
ner_results = nlp(example)
print(ner_results)
Downloads last month
13
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.