orai-nlp/basqueGLUE
Updated • 210 • 6
How to use orai-nlp/ElhBERTeu-nerc with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="orai-nlp/ElhBERTeu-nerc") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("orai-nlp/ElhBERTeu-nerc")
model = AutoModelForTokenClassification.from_pretrained("orai-nlp/ElhBERTeu-nerc")ElhBERTeu model finetuned on the NERC dataset from BasqueGlue.
In Domain and Out-of-domain tasks' train and dev datasets were merged for training. Reported performance results are tested on the merged tests dataset after training for 10 epochs (batch size 32, learning rate 3e-5).
Results on test set:
Per class results:
| CLASS | precision | recall | F1 score | support |
|---|---|---|---|---|
| LOC | 89.43% | 87.95% | 88.68 | 1844 |
| MISC | 70.72% | 67.75% | 69.20 | 502 |
| ORG | 80.04% | 84.32% | 82.12 | 1082 |
| PER | 93.52% | 94.57% | 94.04 | 1359 |
| GLOBAL | 86.51% | 86.81% | 86.66 |
Tagset: ["O", "B-ORG", "B-PER", "I-PER", "I-ORG", "B-LOC", "I-LOC", "B-MISC", "I-MISC"]
Finetuning details: 10 epochs, batch size 32, learning rate 3e-5.