Djacon/rubert-tiny2-russian-emotion-detection

Short Description

The _rubert-tiny2-russian-emotion-detection is a fine-tuned [rubert-tiny2](https://huggingface.co/cointegrated/rubert-tiny2) model for multi-label emotion classification task, specifically on Russian texts. Trained on custom [ru-izard-emotions](https://huggingface.co/datasets/Djacon/ru-izard-emotions) dataset, so this model can recognize a spectrum of 9 emotions, including joy, sadness, anger, enthusiasm, surprise, disgust, fear, guilt, shame + neutral (no emotion). Project was inspired by the [Izard's model](https://en.wikipedia.org/wiki/DifferentialEmotions_Scale) of human emotions.

For more information about model, please check Github repository

Training Parameters:

Optimizer: AdamW
Schedule: LambdaLR
Learning Rate: 1e-4
Batch Size: 64
Number Of Epochs: 10

Emotion Categories:

0. Neutral (Нейтрально)
1. Joy (Радость)
2. Sadness (Грусть)
3. Anger (Гнев)
4. Enthusiasm (Интерес)
5. Surprise (Удивление)
6. Disgust (Отвращение)
7. Fear (Страх)
8. Guilt (Вина)
9. Shame (Стыд)

Test results:

Citations

@misc{Djacon,
  author={Djacon},
  year={2023},
  publisher={Hugging Face},
  journal={Hugging Face Hub},
}