mascIT's picture
Create README.md
f546459
|
raw
history blame
727 Bytes
---
language:
- it
pipeline_tag: text-classification
library_name: transformers
widget:
- example_title: "Esempio 1"
text: "colore"
- example_title: "Esempio 2"
text: "rosso"
- example_title: "Esempio 3"
text: "piuttosto"
---
This model (based on [bert-tiny-ita](https://huggingface.co/mascIT/bert-tiny-ita)) has been finetuned for a lemma classification task, using high quality dictionary data.
Labels:
- 'AGG': 0,
- 'VERBO_INTRANSITIVO': 1,
- 'VERBO_TRANSITIVO': 2,
- 'SOST_MASCHILE': 3,
- 'SOST_FEMMINILE': 4,
- 'AVVERBIO': 5,
- 'AGG_SOSTANTIVATO': 6
The project is still a work in progress, new versions will come with time.
# Training
- epochs: 5
- lr: 1e-4
- optim: AdamW
- weight_decay: 1e-2
# Eval