Update README.md
Browse files
README.md
CHANGED
@@ -29,6 +29,29 @@ The model was trained on the Speech Emotion Recognition dataset from Kaggle, whi
|
|
29 |
- Dataset: Speech Emotion Recognition Dataset (Kaggle)
|
30 |
- Emotions Detected: Sadness, Anger, Disgust, Fear, Happiness, Neutral
|
31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
## How to Use
|
33 |
### Installation
|
34 |
To use this model, you need to install the transformers and torchaudio packages:
|
|
|
29 |
- Dataset: Speech Emotion Recognition Dataset (Kaggle)
|
30 |
- Emotions Detected: Sadness, Anger, Disgust, Fear, Happiness, Neutral
|
31 |
|
32 |
+
## Training Results
|
33 |
+
|
34 |
+
The model achieved the following results on the test set:
|
35 |
+
|
36 |
+
**Test Accuracy**: `0.7435`
|
37 |
+
|
38 |
+
**Classification Report**:
|
39 |
+
|
40 |
+
```plaintext
|
41 |
+
precision recall f1-score support
|
42 |
+
|
43 |
+
sadness 0.68 0.71 0.70 251
|
44 |
+
angry 0.75 0.93 0.83 258
|
45 |
+
disgust 0.86 0.64 0.73 250
|
46 |
+
fear 0.75 0.61 0.67 287
|
47 |
+
happy 0.73 0.68 0.71 231
|
48 |
+
neutral 0.72 0.92 0.81 212
|
49 |
+
|
50 |
+
accuracy 0.74 1489
|
51 |
+
macro avg 0.75 0.75 0.74 1489
|
52 |
+
weighted avg 0.75 0.74 0.74 1489
|
53 |
+
```
|
54 |
+
|
55 |
## How to Use
|
56 |
### Installation
|
57 |
To use this model, you need to install the transformers and torchaudio packages:
|