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:
|