nr: refactor

blink_detection_classifier.py

@@ -161,7 +161,7 @@ plt.show()
 
 # # save the classifier
 # model = {'model': svclassifier, 'scaler': scaler}
-# with open('svclassifier_new3.pkl', 'wb') as fid:
+# with open('svclassifier_new.pkl', 'wb') as fid:
 #     pickle.dump(model, fid)
 
 # rfc = RandomForestClassifier(max_depth=10, random_state=0)

test_file_traverse.py

@@ -23,7 +23,7 @@ def feature_extraction(data_):
 
 
 # load classifier
-with open('svclassifier_new3.pkl', 'rb') as fid:
+with open('svclassifier_new.pkl', 'rb') as fid:
     my_model = pickle.load(fid)
 model = my_model['model']
 scaler = my_model['scaler']