mascIT's picture
Create README.md
f546459
|
raw
history blame
727 Bytes
metadata
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) 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