eriktks/conll2002
Updated • 1.57k • 10
How to use raulgdp/NER-finetuning-BETO-PRO with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="raulgdp/NER-finetuning-BETO-PRO") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("raulgdp/NER-finetuning-BETO-PRO")
model = AutoModelForTokenClassification.from_pretrained("raulgdp/NER-finetuning-BETO-PRO")This model is a fine-tuned version of google-bert/bert-base-uncased on the conll2002 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.1941 | 1.0 | 1041 | 0.1965 | 0.6201 | 0.6836 | 0.6503 | 0.9422 |
| 0.1276 | 2.0 | 2082 | 0.1843 | 0.6666 | 0.7387 | 0.7008 | 0.9487 |
| 0.0885 | 3.0 | 3123 | 0.1760 | 0.7056 | 0.7601 | 0.7319 | 0.9538 |
| 0.0623 | 4.0 | 4164 | 0.1856 | 0.6982 | 0.7670 | 0.7310 | 0.9532 |
| 0.0485 | 5.0 | 5205 | 0.1981 | 0.7018 | 0.7732 | 0.7358 | 0.9536 |
Base model
google-bert/bert-base-uncased