mascIT's picture
Update README.md
19171d2
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

bert-tiny-ita-lemma-classification is a text-classification model (based on bert-tiny-ita) which has been finetuned for a lemma classification task, using a private dataset of high quality dictionary data.

It can classify an italian lemma in the following classes:

  • '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