File size: 1,365 Bytes
46733ad
 
 
4bd6f40
 
 
 
 
 
 
 
 
914bd6b
4bd6f40
914bd6b
4bd6f40
914bd6b
4bd6f40
914bd6b
4bd6f40
914bd6b
4bd6f40
914bd6b
4bd6f40
914bd6b
4bd6f40
 
 
 
 
d5d7cdc
 
 
 
 
 
 
 
 
 
 
 
 
d7d9de1
d5d7cdc
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
---
license: apache-2.0
---
Model description Cased fine-tuned BERT model for English, trained on (manually annotated) Hungarian parliamentary speeches scraped from parlament.hu, and translated with Google Translate API.


Training The fine-tuned version of the original bert-base-cased model (bert-base-cased), trained on HunEmPoli corpus, translated with Google Translate API.


The model can be used as any other (cased) BERT model. It has been tested recognizing emotions at the sentence level in (parliamentary) pre-agenda speeches, where:

'Label_0': Neutral

'Label_1': Fear

'Label_2': Sadness

'Label_3': Anger

'Label_4': Disgust

'Label_5': Success

'Label_6': Joy

'Label_7': Trust


Eval results

                  precision    recall  f1-score   support

           0       1.00      0.50      0.67        46
           1       0.00      0.00      0.00         4
           2       0.70      0.85      0.76       188
           3       0.50      0.09      0.15        11
           4       0.85      0.75      0.80       375
           5       0.78      0.93      0.84       335
           6       0.67      0.28      0.39        36
           7       0.00      0.00      0.00         4

    accuracy                           0.79       999
    macro avg       0.56      0.42     0.45       999
    weighted avg    0.79      0.79     0.77       999