Carlos-Francisco Méndez-Cruz / lcg-bioinfoI-bionlp

Go to a project
Toggle navigation Toggle navigation pinning
Carlos-Francisco Méndez-Cruz
067a9feb 1 parent b402fcd2

Training, crossvalidation and testing dataset

Inline Side-by-side
Showing 1 changed file with 1 additions and 1 deletions
...@@ -221,7 +221,7 @@ if __name__ == "__main__": ...@@ -221,7 +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 - k_range = list(range(3, 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,
......