knn fst commit

knn.py

+import numpy as np
+
+def distance_matrix(a,b):
+    sx = np.sum(a**2, axis=1, keepdims=True)
+    sy = np.sum(b**2, axis=1, keepdims=True)
+    dists = np.sqrt(-2 * a.dot(b.T) + sx + sy.T)
+    return dists
+
+
+
+
+
+if __name__ == "__main__" :
+