模型:

padmalcom/wav2vec2-large-emotion-detection-german

中文

This wav2vec2 based emotion detection model is trained on the emo-DB dataset . Code for training can be found here .

Emotion classes are:

  • 0: 'anger'
  • 1: 'boredom'
  • 2: 'disgust'
  • 3: 'fear'
  • 4: 'happiness'
  • 5: 'sadness'
  • 6: 'neutral'

inference.py shows, how the model can be used.