Model Card
Model Description
This model is a fine-tuned version of shrimp1106/bert-finetuned-ner-accelerate on a custom NER dataset with the following labels:
['O', 'B-PER', 'I-PER', 'B-ORG', 'I-ORG', 'B-LOC', 'I-LOC', 'B-MISC', 'I-MISC']
Intended Use
This model is intended for Named Entity Recognition (NER) tasks.
Training
- Base model:
shrimp1106/bert-finetuned-ner-accelerate
- Epochs: 3
- Batch size: 8
- Optimizer: AdamW
- Learning rate scheduler: linear
Metrics on test set
- Precision: 0.8420
- Recall: 0.7434
- F1 score: 0.7896
- Accuracy: 0.9770
License
CC-BY-4.0 or same as base model (shrimp1106/bert-finetuned-ner-accelerate)
