sklearn中 K近邻 简易使用

K近邻算法是如果一个样本在特征空间的K个最相邻（即特征空间中最邻近）的样本中的大多数属于某一个类别，则该样本也属于这个类别。
他的原理就是求两点之间的距离，看距离谁是最近的，以此来区分我们要预测的这个数据是属于哪个分类。计算距离的公式为欧式距离公式。

from sklearn.neighbors import KNeighborsClassifier
import numpy as np


values = np.array(
    [[0, 0],
    [0.1, 0.2],
    [1, 1],
    [1.1, 0.9]]
)

labels = ['A', 'A', 'B', 'B']

knn = KNeighborsClassifier(n_neighbors=3)
knn.fit(values, labels)
ret = knn.predict([[0.7, 0.8]])
print(ret)

结果是属于B类