diff --git a/knn.py b/knn.py index fece0caf4ff327e411841430f7df1bd0a3cb6b6b..843efe234d45262e4a4ff131500ed796429f3d2b 100644 --- a/knn.py +++ b/knn.py @@ -55,12 +55,15 @@ def bench_knn(): accuracies.append(accuracy) # Save and show the graph of accuracies - fig = plt.figure() + plt.figure(figsize=(8, 6)) + plt.xlabel('K') + plt.ylabel('Accuracy') plt.plot(k_indices, accuracies) plt.title("Accuracy as function of k") + plt.legend() + plt.grid(True) plt.show() plt.savefig('image-classification/results/knn_batch_1.png') - plt.close(fig) if __name__ == "__main__": @@ -75,9 +78,4 @@ if __name__ == "__main__": # x_test = np.array([[1,2],[4,6]]) # x_train = np.array([[2,4],[7,2],[4,6]]) # y_train = [1,2,1] - # dist = distance_matrix(x_test,x_train) - - - - - \ No newline at end of file + # dist = distance_matrix(x_test,x_train) \ No newline at end of file