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