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