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类

上一篇:kNN算法


下一篇:kNN(K- Nearest Neighbor)基本原理