k_nearest.py

import numpy as np


class KNearest:
    X = np.array([])
    y = np.array([])
    k = 0

    def __init__(self, k, X, y):
        self.k = k
        self.X = X
        self.y = y

    def predict(self, x):
        return self.__most__(self.y[np.argsort(np.linalg.norm(self.X - x, axis=1))][0:self.k])

    @staticmethod
    def __most__(arr):
        (values, counts) = np.unique(arr, return_counts=True)
        ind = np.argmax(counts)
        return values[ind]