Edit model card

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

Downloads last month
17
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.