Update README.md
Browse files
README.md
CHANGED
|
@@ -10,11 +10,11 @@ base_model:
|
|
| 10 |
---
|
| 11 |
# Latvian Basic Emotion Classifier
|
| 12 |
|
| 13 |
-
A fine-tuned version of [LVBERT](https://huggingface.co/AiLab-IMCS-UL/lvbert) for multi-label text classification of
|
| 14 |
|
| 15 |
-
The model
|
| 16 |
|
| 17 |
-
|
| 18 |
```yaml
|
| 19 |
0: anger
|
| 20 |
1: disgust
|
|
@@ -25,7 +25,7 @@ Labels predicted by the classifier:
|
|
| 25 |
6: neutral
|
| 26 |
```
|
| 27 |
|
| 28 |
-
The seed used for
|
| 29 |
```python
|
| 30 |
def set_seed(seed=42):
|
| 31 |
random.seed(seed)
|
|
@@ -52,8 +52,8 @@ num_epochs: 3
|
|
| 52 |
|
| 53 |
## Evaluation
|
| 54 |
|
| 55 |
-
Evaluation results on the test split of [go_emotions-lv](https://huggingface.co/datasets/AiLab-IMCS-UL/go_emotions-lv/viewer/simplified_ekman/test)
|
| 56 |
-
| |Precision|Recall|F1-
|
| 57 |
|--------------|---------|------|--------|-------|
|
| 58 |
|anger | 0.57| 0.36| 0.44| 726|
|
| 59 |
|disgust | 0.42| 0.29| 0.35| 123|
|
|
@@ -67,8 +67,8 @@ Evaluation results on the test split of [go_emotions-lv](https://huggingface.co/
|
|
| 67 |
|weighted avg | 0.68| 0.59| 0.63| 5894|
|
| 68 |
|samples avg | 0.62| 0.61| 0.61| 5894|
|
| 69 |
|
| 70 |
-
Evaluation results on the test split of [twitter_emotions-lv](https://huggingface.co/datasets/AiLab-IMCS-UL/twitter_emotions-lv/viewer/simplified_ekman/test)
|
| 71 |
-
| |Precision|Recall|F1-
|
| 72 |
|--------------|---------|------|--------|-------|
|
| 73 |
|anger | 0.94| 0.87| 0.90| 12013|
|
| 74 |
|disgust | 0.92| 0.92| 0.92| 14117|
|
|
|
|
| 10 |
---
|
| 11 |
# Latvian Basic Emotion Classifier
|
| 12 |
|
| 13 |
+
A fine-tuned version of [LVBERT](https://huggingface.co/AiLab-IMCS-UL/lvbert) for multi-label text classification of six basic emotions (+neutral) in Latvian, as defined by Ekman’s theory.
|
| 14 |
|
| 15 |
+
The model is trained on a combined dataset of [go_emotions-lv](https://huggingface.co/datasets/AiLab-IMCS-UL/go_emotions-lv) and [twitter_emotions-lv](https://huggingface.co/datasets/AiLab-IMCS-UL/twitter_emotions-lv).
|
| 16 |
|
| 17 |
+
Predicted labels:
|
| 18 |
```yaml
|
| 19 |
0: anger
|
| 20 |
1: disgust
|
|
|
|
| 25 |
6: neutral
|
| 26 |
```
|
| 27 |
|
| 28 |
+
The random seed used for initialization was 42:
|
| 29 |
```python
|
| 30 |
def set_seed(seed=42):
|
| 31 |
random.seed(seed)
|
|
|
|
| 52 |
|
| 53 |
## Evaluation
|
| 54 |
|
| 55 |
+
Evaluation results on the test split of [go_emotions-lv](https://huggingface.co/datasets/AiLab-IMCS-UL/go_emotions-lv/viewer/simplified_ekman/test):
|
| 56 |
+
| |Precision|Recall|F1-score|Support|
|
| 57 |
|--------------|---------|------|--------|-------|
|
| 58 |
|anger | 0.57| 0.36| 0.44| 726|
|
| 59 |
|disgust | 0.42| 0.29| 0.35| 123|
|
|
|
|
| 67 |
|weighted avg | 0.68| 0.59| 0.63| 5894|
|
| 68 |
|samples avg | 0.62| 0.61| 0.61| 5894|
|
| 69 |
|
| 70 |
+
Evaluation results on the test split of [twitter_emotions-lv](https://huggingface.co/datasets/AiLab-IMCS-UL/twitter_emotions-lv/viewer/simplified_ekman/test):
|
| 71 |
+
| |Precision|Recall|F1-score|Support|
|
| 72 |
|--------------|---------|------|--------|-------|
|
| 73 |
|anger | 0.94| 0.87| 0.90| 12013|
|
| 74 |
|disgust | 0.92| 0.92| 0.92| 14117|
|