eriktks/conll2003
Updated • 22.9k • 174
How to use srvmishra832/CoNLL2003_NER_BERT_Base_Cased with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="srvmishra832/CoNLL2003_NER_BERT_Base_Cased") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("srvmishra832/CoNLL2003_NER_BERT_Base_Cased")
model = AutoModelForTokenClassification.from_pretrained("srvmishra832/CoNLL2003_NER_BERT_Base_Cased", device_map="auto")This model is a fine-tuned version of bert-base-cased on the conll2003 dataset. It achieves the following results on the evaluation set:
BERT Base Cased Model available at: bert-base-cased
Named Entity Recognition Task (English)
CONLL2023 Dataset available at: conll2003 dataset
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|---|
| 0.298 | 1.0 | 220 | 0.0792 | 0.8463 | 0.8995 | 0.8721 | 0.9769 |
| 0.0647 | 2.0 | 440 | 0.0617 | 0.9088 | 0.9362 | 0.9223 | 0.9830 |
| 0.0394 | 3.0 | 660 | 0.0574 | 0.9207 | 0.9443 | 0.9324 | 0.9846 |
| 0.0286 | 4.0 | 880 | 0.0559 | 0.9195 | 0.9438 | 0.9315 | 0.9855 |
| 0.0222 | 5.0 | 1100 | 0.0571 | 0.9250 | 0.9458 | 0.9353 | 0.9855 |
Base model
google-bert/bert-base-cased