Hugging Face's logo

seara
/
rubert-tiny2-russian-emotion-detection-cedr

Text Classification
Transformers
PyTorch
TensorFlow
JAX
Safetensors
Russian
bert
sentiment-analysis
multi-label-classification
sentiment analysis
rubert
sentiment
tiny
russian
multilabel
classification
emotion-classification
emotion-recognition
emotion
emotion-detection
Inference Endpoints
Model card Files Community
1
seara commited on
Commit
6ebb6d7
•
1 Parent(s): c9a274b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -10
README.md CHANGED
@@ -9,14 +9,18 @@ metrics:
9
  - recall
10
  pipeline_tag: text-classification
11
  tags:
12
- - rubert
13
- - emotion
14
- - sentiment
15
  - emotion-classification
16
  - emotion-recognition
 
 
17
  - russian
 
 
 
 
18
  - classification
19
  - multilabel
 
20
  datasets:
21
  - cedr
22
  ---
@@ -33,6 +37,17 @@ The task is a __multi-label classification__ with the following labels:
33
  5: anger
34
  ```
35
 
 
 
 
 
 
 
 
 
 
 
 
36
  ## Usage
37
 
38
  ```python
@@ -63,10 +78,10 @@ num_epochs: 30
63
 
64
  ## Eval results (on test split)
65
 
66
- | |no_emotion|joy |sadness|surprise|fear |anger |micro avg|macro avg|weighted avg|samples avg|
67
- |---------|----------|------|-------|--------|-------|------|---------|---------|------------|-----------|
68
- |precision|0.8176 |0.8371|0.8425 |0.7902 |0.7833 |0.5467|0.811 |0.7696 |0.8034 |0.7811 |
69
- |recall |0.8365 |0.83 |0.847 |0.6647 |0.6667 |0.328 |0.776 |0.6955 |0.776 |0.7792 |
70
- |f1-score |0.8269 |0.8336|0.8447 |0.722 |0.7203 |0.41 |0.7931 |0.7263 |0.787 |0.7788 |
71
- |support |734.0 |353.0 |379.0 |170.0 |141.0 |125.0 |1902.0 |1902.0 |1902.0 |1902.0 |
72
- |auc-roc |0.9241 |0.9649|0.9557 |0.913 |0.9118 |0.7732|0.9355 |0.9071 |0.9261 | |
 
9
  - recall
10
  pipeline_tag: text-classification
11
  tags:
 
 
 
12
  - emotion-classification
13
  - emotion-recognition
14
+ - multi-label-classification
15
+ - emotion
16
  - russian
17
+ - rubert
18
+ - tiny
19
+ - sentiment
20
+ - sentiment-analysis
21
  - classification
22
  - multilabel
23
+ - bert
24
  datasets:
25
  - cedr
26
  ---
 
37
  5: anger
38
  ```
39
 
40
+ Label to Russian label:
41
+
42
+ ```yaml
43
+ no_emotion: нет эмоции
44
+ joy: радость
45
+ sadness: грусть
46
+ surprise: удивление
47
+ fear: страх
48
+ anger: злость
49
+ ```
50
+
51
  ## Usage
52
 
53
  ```python
 
78
 
79
  ## Eval results (on test split)
80
 
81
+ | |no_emotion|joy |sadness|surprise|fear |anger|micro avg|macro avg|weighted avg|
82
+ |---------|----------|------|-------|--------|-------|-----|---------|---------|------------|
83
+ |precision|0.82 |0.84 |0.84 |0.79 |0.78 |0.55 |0.81 |0.77 |0.8 |
84
+ |recall |0.84 |0.83 |0.85 |0.66 |0.67 |0.33 |0.78 |0.7 |0.78 |
85
+ |f1-score |0.83 |0.83 |0.84 |0.72 |0.72 |0.41 |0.79 |0.73 |0.79 |
86
+ |auc-roc |0.92 |0.96 |0.96 |0.91 |0.91 |0.77 |0.94 |0.91 |0.93 |
87
+ |support |734 |353 |379 |170 |141 |125 |1902 |1902 |1902 |