Update main.py

main.py

@@ -14,7 +14,7 @@ batch_path = "data/cifar-10-python\cifar-10-batches-py"
 data, labels = read_cifar.read_cifar(batch_path)
 data_train, labels_train, data_test, labels_test = read_cifar.split_dataset(data, labels, split)
 
-"""k_values = range(1, 21)
+k_values = range(1, 21)
 accuracies = []
 times = []
 
@@ -45,7 +45,7 @@ plt.ylabel('time')
 plt.xticks(k_values)
 plt.grid(True)
 plt.savefig('results/time_knn.png')
-plt.show()"""
+plt.show()
 
 
 train_accuracies,test_accuracy =  mlp.run_mlp_training(data_train, labels_train, data_test, labels_test, d_h, learning_rate, num_epochs)