fix knn default cmd argumment bug

mesarcik · Dec 9, 2022 · 3b830c9 · 3b830c9
1 parent a67f651
commit 3b830c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/cmd_input.py b/utils/cmd_input.py
@@ -24,7 +24,7 @@
 parser.add_argument('-radius', metavar='-r', type=float, nargs='+', default=[0.1,0.5,1,2,5,10],
                     help = 'The radius of the unit circle for finding neighbours in frNN')
 parser.add_argument('-algorithm', metavar='-nn', type=str, choices={"frnn", "knn"}, 
-                    default='frnn', help = 'The algorithm for calculating neighbours')
+                    default='knn', help = 'The algorithm for calculating neighbours')
 parser.add_argument('-data', metavar='-d', type=str, default='MNIST',
                     help = 'The dataset for training and testing the model on')