antonjaragon commited on
Commit
2e69023
1 Parent(s): f32a5f7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -4
README.md CHANGED
@@ -15,22 +15,31 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # emotions_6_classes_small
17
 
18
- This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the None dataset.
 
 
19
  It achieves the following results on the evaluation set:
20
  - Loss: 0.9106
21
  - Accuracy: 0.7920
22
 
23
  ## Model description
24
 
25
- More information needed
 
 
 
 
 
 
26
 
27
  ## Intended uses & limitations
28
 
29
- More information needed
30
 
31
  ## Training and evaluation data
32
 
33
- More information needed
 
34
 
35
  ## Training procedure
36
 
 
15
 
16
  # emotions_6_classes_small
17
 
18
+ This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the 'Audio emotions' public dataset, available form https://www.kaggle.com/datasets/uldisvalainis/audio-emotions.
19
+ 'Surprised' class was discarded due to lack of samples.
20
+
21
  It achieves the following results on the evaluation set:
22
  - Loss: 0.9106
23
  - Accuracy: 0.7920
24
 
25
  ## Model description
26
 
27
+ Classifies audios into 6 emotions:
28
+ - Angry
29
+ - Happy
30
+ - Sad
31
+ - Neutral
32
+ - Fearful
33
+ - Disgusted
34
 
35
  ## Intended uses & limitations
36
 
37
+ This model was trained for educational purposes.
38
 
39
  ## Training and evaluation data
40
 
41
+ - Training: 80%
42
+ - Test: 20%
43
 
44
  ## Training procedure
45