---
license: mit
language:
- ru
- en
tags:
- russian
- classification
- emotion
- emotion-detection
- emotion-recognition
- multiclass
widget:
- text: Сейчас ровно час дня
- text: Сегодня такой замечательный день!
- text: Жалею, что вчера сходил на этот концерт
- text: Что за бред я только что посмотрел...
- text: Куда бы сегодня сходить?
- text: Воу, это было так неожиданно
- text: Фу, эта еда просто отвратительна!
- text: В темной комнате услышал тихий посторонний шорох
- text: Извини, я не хотел чтобы так все произошло
datasets:
- Djacon/ru-izard-emotions
---

## 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/Differential_Emotions_Scale) of human emotions.

For more information about model, please check [Github repository](https://github.com/Djacon/russian-emotion-detection)

## Training Parameters:
```yaml
Optimizer: AdamW
Schedule: LambdaLR
Learning Rate: 1e-4
Batch Size: 64
Number Of Epochs: 10
```

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

## Test results:

||Neutral|Joy|Sadness|Anger|Enthusiasm|Surprise|Disgust|Fear|Guilt|Shame|Mean|
|-|-|-|-|-|-|-|-|-|-|-|-|
|AUC|0.7319|0.8234|0.8069|0.7884|0.8493|0.8047|0.8147|0.9034|0.8528|0.7145|0.8090|
|F1 micro|0.7192|0.7951|0.8204|0.7642|0.8630|0.9032|0.9156|0.9482|0.9526|0.9606|0.8642|
|F1 macro|0.6021|0.7237|0.6548|0.6274|0.7291|0.5712|0.4780|0.8158|0.4879|0.4900|0.6180|


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