r1char9's picture
Update README.md
5125641 verified
metadata
license: mit
language:
  - ru
pipeline_tag: text-classification
tags:
  - sentiment-analysis
  - multi-label-classification
  - sentiment analysis
  - rubert
  - sentiment
  - bert
  - tiny
  - russian
  - multilabel
  - classification
  - emotion-classification
  - emotion-recognition
  - emotion

Модель RuBERT-tiny2 была fine-tuned для задачи emotion classification, предназначенная для Russian текст. Выполняет задачу multi-label classification с слудующимим категориями:

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

Категории для русского языка:

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

from transformers import pipeline
model = pipeline(model="r1char9/rubert-tiny2-ru-go-emotions")
model("Привет, ты мне нравишься!")
# [{'label': 'love', 'score': 0.5955629944801331}]