unimelb-nlp/wikiann
Viewer • Updated • 2M • 16.5k • 124
How to use mmaguero/wikiann-xlm-v-base with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="mmaguero/wikiann-xlm-v-base") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("mmaguero/wikiann-xlm-v-base")
model = AutoModelForTokenClassification.from_pretrained("mmaguero/wikiann-xlm-v-base", device_map="auto")This model is a fine-tuned version of facebook/xlm-v-base on the Spanish and Guarani sets of unimelb-nlp/wikiann dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|---|
| 0.255 | 1.0 | 1764 | 0.1951 | 0.8888 | 0.9017 | 0.8952 | 0.9465 |
| 0.1882 | 2.0 | 3528 | 0.1656 | 0.9014 | 0.9112 | 0.9063 | 0.9544 |
| 0.1454 | 3.0 | 5292 | 0.1599 | 0.9070 | 0.9191 | 0.9130 | 0.9575 |
Base model
facebook/xlm-v-base