KNN是一种常见的监督学习算法，工作机制很好理解：给定测试样本，基于某种距离度量找出训练集中与其最靠近的k个训练样本，然后基于这k个“邻居”的信息来进行预测。总结一句话就是“近朱者赤，近墨者黑”。
KNN可用作分类也可用于回归，在分类任务中可使用“投票法”，即选择这k个样本中出现最多的类别标记作为测试结果；在回归任务中可使用“平均法”将这k个样本的标记平均值作为预测结果；还可以基于距离远近进行加权平均或加权投票，距离越近的样本权重越大。
KNN和之前介绍的监督学习算法有一个很大的不同，它没有前期的训练过程，是一种“懒惰学习”的算法，只有收到测试样本后，再和训练样本进行比较处理。
初学者容易把KNN和K-means搞混淆，虽然都有K，：-）但这是两种不同的算法，二者区别如下：
KNN K-Means
不同点 是一种分类算法，属于监督学习的范畴，训练数据是带有label的 是一种聚类算法，属于非监督学习的范畴，训练数据没有label，杂乱无章的
没有明显的训练过程，属于lazy learning 有明确的训练过程
K的含义：与预测样本距离最近的K个样本 K的含义：K是事前人工定好的参数，假设数据集可分为K个簇
相同点 ：都用到了NN（nearst Neighbor）算法，一般用KD树来实现。
--KNN算法基本原理
KNN算法简单的步骤如下：
（1）计算距离：给定测试对象，计算它与训练集中每个对象的距离，空间距离的计算方法有多种，有欧式距离、夹角余弦（多在文本分类中使用）等。
（2）找邻居：圈定距离最近的k个对象，作为测试对象的近邻。
（3）做分类：根据这k个近邻归属的主要类别，对测试对象进行分类。
下面通过一个简单的示例说明下KNN算法是怎么进行分类的：
上图的蓝色方块和红色三角是已经打好label的数据，绿色圆圈是待分类的测试数据。
如果我们让K=3，那么上图实心圆圈中的两个三角和一个方块就是离测试数据最近的3个点，那么通过投票法则，测试数据会被分类为红色三角；
如果我们让K=5，那么上图虚线圆圈中的两个三角和三个方块就是离测试数据最近的5个点，通过投票法则，测试数据则会被分类为蓝色方块；
整个算法的原理是不是很简单？但实际上并没有那么简单，K如何选择？数据之间的距离怎么计算？
--K值的选择
如果K值太小，整体模型会变得复杂，容易发生过拟合，容易将一些噪声学习进来，二忽略数据的真实分布。
如果K值过大，模型会变得相对简单，可以减少学习的估计误差，但近似误差会变大，比如极端情况下K=N（N维训练样本数），则不论预测对象是什么，预测结果都将是训练集中最多的类型，这显然是一个过渡简化的模型，无法实际应用。
k值一般采用交叉验证或者Grid Search的方法确定。
--距离计算
提取数据的特征值，根据特征值组成一个n维实数向量空间（特征空间），然后计算向量之间的空间距离，如欧式距离、余弦相似度等。
对于数据
和
，其特征空间为n维实数向量空间：，
欧式距离计算公式为：
余弦相似度计算公式为：
余弦相似度的值越接近1表示其越相似，接近0表示其差异越大。余弦相似度更多应用在文本类任务中。
--代码示例
依旧以sklearn中的cancer数据集为例，做一个通过30维特征判断是否患癌症的示例，示例中数据量很少，只有569条数据，每条数据各有30个特征数值。采用sklearn中的KNN分类器，除k外都采用默认参数，距离度量采用欧式距离 。通过交叉验证法来确定最佳的K值，从下图可见，K=14时，验证准确率最高。
1 __author__ = 'z00421185' 2 3 import pandas as pd 4 from sklearn import datasets 5 import matplotlib.pyplot as plt 6 from sklearn.model_selection import cross_val_score 7 from sklearn.model_selection import train_test_split 8 from sklearn.metrics import accuracy_score 9 from sklearn.neighbors import KNeighborsClassifier 10 11 breast_data = datasets.load_breast_cancer() 12 data = pd.DataFrame(datasets.load_breast_cancer().data) 13 data.columns = breast_data['feature_names'] 14 15 data_np = breast_data['data'] 16 target_np = breast_data['target'] 17 print(data_np.shape) 18 19 x_train, x_test, y_train, y_test = train_test_split(data_np, target_np, test_size=0.3, random_state=0) 20 21 # 设定交叉验证k的范围，一般从1~样本数的开方 22 k_range = range(1, 24) 23 scores = [] 24 for k in k_range: 25 knn = KNeighborsClassifier(k, metric='euclidean') 26 score = cross_val_score(knn, x_train, y_train, cv=10, scoring='accuracy') 27 scores.append(score.mean()) 28 29 # 从折线图上看最佳K取值 30 plt.plot(k_range, scores) 31 plt.xlabel('K') 32 plt.ylabel('Accuracy') 33 plt.show() 34 35 model = KNeighborsClassifier(n_neighbors=13) 36 model.fit(x_train, y_train) 37 y_pred = model.predict(x_test) 38 print(accuracy_score(y_test, y_pred)) 39 --------------------------------- 40 0.9649122807017544
作者：华为云专家 周捷
HDC.Cloud 华为开发者大会2020 即将于2020年2月11日-12日在深圳举办，是一线开发者学习实践鲲鹏通用计算、昇腾AI计算、数据库、区块链、云原生、5G等ICT开放能力的最佳舞台。
欢迎报名参会（https://www.huaweicloud.com/HDC.Cloud.html?utm_source=&utm_medium=&utm_campaign=&utm_content=techcommunity）
