From 69c44ac44367e95f9b61b2aaceab2e8ac6a5506e Mon Sep 17 00:00:00 2001
From: lucile <lucile.audard@ecl20.ec-lyon.fr>
Date: Mon, 23 Oct 2023 09:22:47 +0200
Subject: [PATCH] Create knn.py

---
 knn.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 knn.py

diff --git a/knn.py b/knn.py
new file mode 100644
index 0000000..043e7ca
--- /dev/null
+++ b/knn.py
@@ -0,0 +1,36 @@
+import numpy as np
+
+
+def distance_matrix(mat1, mat2):
+    dists = np.sqrt(np.matmul(mat1, mat1)+ np.matmul(mat2, mat2) - 2 * np.matmul(mat1, mat2))
+    return dists
+
+def knn_predict(dists, labels_train, k):
+    return predicted_labels
+
+
+
+
+
+
+
+
+mat1 = np.array([[1, 2],
+                 [3, 4]])
+
+mat2 = np.array([[5, 6],
+                 [7, 8]])
+
+A = np.matmul(mat1, mat1)
+print(A)
+
+B = np.matmul(mat2, mat2)
+print(B)
+
+C = 2 * np.matmul(mat1, mat2)
+print(C)
+
+print(A + B - C)
+
+mat = distance_matrix(mat1, mat2)
+print(mat)
\ No newline at end of file
-- 
GitLab