Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

...@@ -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()
......