Training, crossvalidation and testing dataset

@@ -221,7 +221,11 @@ if __name__ == "__main__":
                                                           cv=crossV, n_jobs=jobs, verbose=3)
                                                           cv=crossV, n_jobs=jobs, verbose=3)
     elif args.classifier == 'kNN':
     elif args.classifier == 'kNN':
         # kNN
         # kNN
-        myClassifier = KNeighborsClassifier()
+        k_range = list(range(3, 7, 2))
+        classifier = KNeighborsClassifier()
+        paramGrid = {'n_neighbors ': k_range}
+        myClassifier = model_selection.RandomizedSearchCV(classifier, paramGrid, n_iter=nIter,
+                                                          cv=crossV, n_jobs=jobs, verbose=3)
     else:
     else:
         print("Bad classifier")
         print("Bad classifier")
         exit()
         exit()