File size: 2,544 Bytes
c50c06e 1161779 6c15f79 a9218a7 1161779 6c15f79 a9218a7 6c15f79 c50c06e a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 6c15f79 a9218a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
---
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},
}
``` |