diff --git a/machine-learning/kn.py b/machine-learning/kn.py new file mode 100644 index 0000000..20770f9 --- /dev/null +++ b/machine-learning/kn.py @@ -0,0 +1,20 @@ +def dist(x1,x2): + return np.sqrt(sum((x1-x2)**2)) + +# Test Time +def knn(X,Y,queryPoint,k=5): + + vals = [] + m = X.shape[0] + + for i in range(m): + d = dist(queryPoint,X[i]) + vals.append((d,Y[i])) + + new_vals = np.unique(vals[:,1],return_counts=True) + + + index = new_vals[1].argmax() + pred = new_vals[0][index] + + return pred