Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

...@@ -221,10 +221,7 @@ if __name__ == "__main__": ...@@ -221,10 +221,7 @@ if __name__ == "__main__":
221 cv=crossV, n_jobs=jobs, verbose=3) 221 cv=crossV, n_jobs=jobs, verbose=3)
222 elif args.classifier == 'kNN': 222 elif args.classifier == 'kNN':
223 # kNN 223 # kNN
224 - classifier = KNeighborsClassifier() 224 + myClassifier = KNeighborsClassifier()
225 - paramGrid = {'n_neighbors ': [3, 5, 7]}
226 - myClassifier = model_selection.GridSearchCV(classifier, paramGrid,
227 - cv=crossV, n_jobs=jobs, verbose=3)
228 else: 225 else:
229 print("Bad classifier") 226 print("Bad classifier")
230 exit() 227 exit()
......