bug fix in test.py

Author: x4nth055

emotion_recognition.py

@@ -5,12 +5,6 @@
 from sklearn.metrics import accuracy_score, make_scorer, fbeta_score, mean_squared_error, mean_absolute_error
 from sklearn.metrics import confusion_matrix
 from sklearn.model_selection import GridSearchCV
-from sklearn.svm import SVC
-from sklearn.ensemble import AdaBoostClassifier, GradientBoostingClassifier, RandomForestClassifier
-from sklearn.naive_bayes import BernoulliNB, GaussianNB, ComplementNB
-from sklearn.tree import DecisionTreeClassifier
-from sklearn.neighbors import KNeighborsClassifier
-from sklearn.neural_network import MLPClassifier
 
 import matplotlib.pyplot as pl
 from time import time

test.py

@@ -3,7 +3,6 @@
 import pyaudio
 import os
 import wave
-import numpy as np
 from sys import byteorder
 from array import array
 from struct import pack
@@ -128,15 +127,6 @@ def get_estimators_name(estimators):
     return ','.join(result), {estimator_name.strip('"'): estimator for estimator_name, (estimator, _, _) in zip(result, estimators)}
 
 
-audio_config = {
-    "mfcc": True,
-    "chroma": True,
-    "mel": True,
-    "contrast": False,
-    "tonnetz": False,
-}
-
-
 
 if __name__ == "__main__":
     estimators = get_best_estimators(True)
@@ -161,7 +151,8 @@ def get_estimators_name(estimators):
     # Parse the arguments passed
     args = parser.parse_args()
 
-    detector = EmotionRecognizer(estimator_dict[args.model], emotions=args.emotions.split(","), audio_config=audio_config, verbose=0)
+    features = ["mfcc", "chroma", "mel"]
+    detector = EmotionRecognizer(estimator_dict[args.model], emotions=args.emotions.split(","), features=features, verbose=0)
     detector.train()
     print("Test accuracy score: {:.3f}%".format(detector.test_score()*100))
     print("Please talk")