|
--- |
|
license: mit |
|
language: |
|
- ru |
|
metrics: |
|
- f1 |
|
- roc_auc |
|
- precision |
|
- recall |
|
pipeline_tag: text-classification |
|
tags: |
|
- sentiment-analysis |
|
- multi-label-classification |
|
- sentiment analysis |
|
- rubert |
|
- sentiment |
|
- bert |
|
- tiny |
|
- russian |
|
- multilabel |
|
- classification |
|
- emotion-classification |
|
- emotion-recognition |
|
- emotion |
|
datasets: |
|
- seara/ru_go_emotions |
|
--- |
|
|
|
This is [RuBERT-tiny2](https://huggingface.co/cointegrated/rubert-tiny2) model fine-tuned for __emotion classification__ of short __Russian__ texts. |
|
The task is a __multi-label classification__ with the following labels: |
|
|
|
```yaml |
|
0: admiration |
|
1: amusement |
|
2: anger |
|
3: annoyance |
|
4: approval |
|
5: caring |
|
6: confusion |
|
7: curiosity |
|
8: desire |
|
9: disappointment |
|
10: disapproval |
|
11: disgust |
|
12: embarrassment |
|
13: excitement |
|
14: fear |
|
15: gratitude |
|
16: grief |
|
17: joy |
|
18: love |
|
19: nervousness |
|
20: optimism |
|
21: pride |
|
22: realization |
|
23: relief |
|
24: remorse |
|
25: sadness |
|
26: surprise |
|
27: neutral |
|
``` |
|
|
|
Label to Russian label: |
|
|
|
```yaml |
|
admiration: восхищение |
|
amusement: веселье |
|
anger: злость |
|
annoyance: раздражение |
|
approval: одобрение |
|
caring: забота |
|
confusion: непонимание |
|
curiosity: любопытство |
|
desire: желание |
|
disappointment: разочарование |
|
disapproval: неодобрение |
|
disgust: отвращение |
|
embarrassment: смущение |
|
excitement: возбуждение |
|
fear: страх |
|
gratitude: признательность |
|
grief: горе |
|
joy: радость |
|
love: любовь |
|
nervousness: нервозность |
|
optimism: оптимизм |
|
pride: гордость |
|
realization: осознание |
|
relief: облегчение |
|
remorse: раскаяние |
|
sadness: грусть |
|
surprise: удивление |
|
neutral: нейтральность |
|
``` |
|
|
|
## Usage |
|
|
|
```python |
|
from transformers import pipeline |
|
model = pipeline(model="r1char9/rubert-tiny2-ru-go-emotions") |
|
model("Привет, ты мне нравишься!") |
|
# [{'label': 'love', 'score': 0.5955629944801331}] |
|
model('Hello, I love you.') |
|
# [{'label': 'love', 'score': 0.911257266998291}] |
|
``` |
|
|
