fixed bug on predict_proba() method of EmotionRecognizer

x4nth055 · x4nth055 · commit 9538048f2e57 · 2020-09-02T18:41:29.000+02:00
diff --git a/emotion_recognition.py b/emotion_recognition.py
@@ -182,7 +182,7 @@ def predict_proba(self, audio_path):
             feature = extract_feature(audio_path, **self.audio_config).reshape(1, -1)
             proba = self.model.predict_proba(feature)[0]
             result = {}
-            for emotion, prob in zip(self.emotions, proba):
+            for emotion, prob in zip(self.model.classes_, proba):
                 result[emotion] = prob
             return result
         else:
