diff --git a/knn.py b/knn.py
new file mode 100644
index 0000000000000000000000000000000000000000..043e7ca81258066d76c9d9e0c509994504739109
--- /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