Showing
1 changed file
with
1 additions
and
1 deletions
... | @@ -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") | ... | ... |
-
Please register or login to post a comment