eriktks/conll2003
Updated • 32.7k • 169
How to use knoam/bert-finetuned-ner with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="knoam/bert-finetuned-ner") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("knoam/bert-finetuned-ner")
model = AutoModelForTokenClassification.from_pretrained("knoam/bert-finetuned-ner")This model is a fine-tuned version of bert-base-cased on the conll2003 dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|---|
| 0.0751 | 1.0 | 1756 | 0.0709 | 0.8998 | 0.9310 | 0.9151 | 0.9805 |
| 0.0341 | 2.0 | 3512 | 0.0729 | 0.9336 | 0.9467 | 0.9401 | 0.9844 |
| 0.0227 | 3.0 | 5268 | 0.0666 | 0.9326 | 0.9497 | 0.9410 | 0.9858 |
Base model
google-bert/bert-base-cased