Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

...@@ -224,7 +224,7 @@ if __name__ == "__main__": ...@@ -224,7 +224,7 @@ if __name__ == "__main__":
224 k_range = list(range(3, 7, 2)) 224 k_range = list(range(3, 7, 2))
225 classifier = KNeighborsClassifier() 225 classifier = KNeighborsClassifier()
226 paramGrid = {'n_neighbors ': k_range} 226 paramGrid = {'n_neighbors ': k_range}
227 - myClassifier = model_selection.RandomizedSearchCV(classifier, paramGrid, n_iter=nIter, 227 + myClassifier = model_selection.RandomizedSearchCV(classifier, paramGrid, n_iter=3,
228 cv=crossV, n_jobs=jobs, verbose=3) 228 cv=crossV, n_jobs=jobs, verbose=3)
229 else: 229 else:
230 print("Bad classifier") 230 print("Bad classifier")
......