From 3b830c9955dbfc023ad5d1c3a1f9b940b6845829 Mon Sep 17 00:00:00 2001 From: Misha Mesarcik Date: Fri, 9 Dec 2022 16:38:48 +0100 Subject: [PATCH] fix knn default cmd argumment bug --- utils/cmd_input.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/cmd_input.py b/utils/cmd_input.py index 977449a..41aa49f 100644 --- 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')