轻松玩转 Scikit-Learn 系列 —— KNN 算法

scikit-learn 是最受欢迎的机器学习库之一，它提供了各种主流的机器学习算法的API接口供使用者调用，让使用者可以方便快捷的搭建一些机器学习模型，并且通过调参可以达到很高的准确率。

这次我们主要介绍scikit-learn中k近邻算法（以下简称为KNN）的使用。

KNN是一种非参数机器学习算法（机器学习中通过模型训练而学到的是模型参数，而要人工调整的是超参数，请注意避免混淆）。使用KNN首先要有一个已知的数据集D，数据集内对于任意一个未知标签的样本数据x，可以通过计算x与D中所有样本点的距离，取出与x距离最近的前k个已知数据，用该k个已知数据的标签对x进行投票，哪一类票数最多，x就是哪一类，这是kNN的大概思想，以下举个例子方便理解。

正方形该分到哪个类？

在上图中有2个已知类别——红色五角星和蓝色三角形和一个未知样本——绿色方格。现在我们要用KNN算法对绿色方格进行分类，以判定其属于这两类中的哪一类，首先令k=5，通过计算距离我们可以知道距离绿色方格最近的5个样本中（假设绿色方格位于圆心），有2个红色五角星，3个蓝色三角形。通过投票可知：蓝色三角形得3票，红色五角星得2票，因此绿色方格应该属于蓝色三角形。kNN就是这样工作的。

原文链接