Showing
1 changed file
with
7 additions
and
7 deletions
... | @@ -219,13 +219,13 @@ if __name__ == "__main__": | ... | @@ -219,13 +219,13 @@ if __name__ == "__main__": |
219 | paramGrid = {'alpha': scipy.stats.expon(scale=1.0)} | 219 | paramGrid = {'alpha': scipy.stats.expon(scale=1.0)} |
220 | myClassifier = model_selection.RandomizedSearchCV(classifier, paramGrid, n_iter=nIter, | 220 | myClassifier = model_selection.RandomizedSearchCV(classifier, paramGrid, n_iter=nIter, |
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 | - k_range = list(range(1, 7, 2)) | 224 | + # k_range = list(range(1, 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=3, | 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") |
231 | exit() | 231 | exit() | ... | ... |
-
Please register or login to post a comment