Text Classification
Transformers
Safetensors
Latvian
bert
text-embeddings-inference
normundsg commited on
Commit
5b52e67
·
verified ·
1 Parent(s): 68d510b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
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 6 base emotions (+neutral) in Latvian, as per Dr. Ekman theory.
14
 
15
- The model was trained on the combined [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) datasets.
16
 
17
- Labels predicted by the classifier:
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 the random number generator - 42:
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-Score|Support|
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-Score|Support|
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|