KELONMYOSA's picture
Update README.md
7735ba6
|
raw
history blame
881 Bytes
metadata
license: apache-2.0
datasets:
  - KELONMYOSA/dusha_emotion_audio
language:
  - ru
pipeline_tag: audio-classification
metrics:
  - accuracy

Speech Emotion Recognition

The model is a fine-tuned version of facebook/wav2vec2-xls-r-300m for a Speech Emotion Recognition (SER) task.

The dataset used to fine-tune the original pre-trained model is the DUSHA dataset. The dataset consists of about 125 000 audio recordings in Russian with four basic emotions that usually appear in a dialog with a virtual assistant: Happiness (Positive), Sadness, Anger and Neutral emotion.

emotions = ['neutral', 'positive', 'angry', 'sad', 'other']

It achieves the following results:

  • Training Loss: 0.528700
  • Validation Loss: 0.349617
  • Accuracy: 0.901369