diff --git a/knn.py b/knn.py new file mode 100644 index 0000000000000000000000000000000000000000..cffe5125c3a8152de1021aef57066500dac0d101 --- /dev/null +++ b/knn.py @@ -0,0 +1,14 @@ +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__" : +