---
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](https://huggingface.co/mascIT/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