Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

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