From 7460daf410ee0af84cb99ac20ae20e659222dcee Mon Sep 17 00:00:00 2001 From: Dubray Chloe <chloe.dubray@etu.ec-lyon.fr> Date: Sat, 4 Nov 2023 21:39:21 +0000 Subject: [PATCH] Upload New File --- knn.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 knn.py diff --git a/knn.py b/knn.py new file mode 100644 index 0000000..aa5b8ef --- /dev/null +++ b/knn.py @@ -0,0 +1,19 @@ +import numpy as np +import pickle +import os + + +def distance_matrix (M1, M2) : + sum_squares_1 = np.sum(M1**2, axis = 1, keepdims = True) + sum_squares_2 = np.sum(M2**2, axis = 1, keepdims = True) + + dot_product = np.dot(M1, M2.T) + dists = np.sqrt(sum_squares_1 - 2*dot_product + sum_squares_2.T) + + return dists + +split_dataset (data, labels, split) + +def knn_predict (dists, labels_train, k) : + train_set + test_set \ No newline at end of file -- GitLab