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时，验证准确率最高。

 

 -Python 代码

 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）