diff --git a/knn.py b/knn.py index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..a322bead56e4cc536487c1fa3a0d4a8eef095854 100644 --- a/knn.py +++ b/knn.py @@ -0,0 +1,8 @@ +def distance_matrix(Y , X): + #This function takes as parameters two matrices X and Y + dists = np.sqrt(np.sum(-2 * np.multiply(X, Y)+ np.multiply(Y, Y) + np.multiply(X, X))) + #dists is the euclidian distance between two matrices + return dists + +def knn_predict(dists, labels_train,k): +