临摹微笑
import numpy as npclass knn: def __init__ (self, x, y, k): self.k = k self.x_data = x self.y_data = y def predict(self, test): dist = np.sqrt(np.sum((self.x_data-test)**2,axis=1)) closest = np.argpartition(dist, self.k)[0:self.k] a,b = np.unique(self.y_data[closest],return_counts=True) return a[np.where(b == b.max())]x:特征,y:标签,k:邻居数量我希望这可以帮助你!