使用KNN进行分类和回归

点击上方“Deephub Imba”,关注公众号,好文章不错过 !

一般情况下k-Nearest Neighbor (KNN)都是用来解决分类的问题，其实KNN是一种可以应用于数据分类和预测的简单算法，本文中我们将它与简单的线性回归进行比较。

KNN模型是一个简单的模型，可以用于回归和分类任务。大部分的机器学习算法都是用它的名字来描述的KNN也是一样，使用一个空间来表示邻居的度量，度量空间根据集合成员的特征定义它们之间的距离。对于每个测试实例，使用邻域来估计响应变量的值。估计可以使用最多k个邻域来进行，超参数控制算法的学习方式;它们不是根据训练数据估计出来的，而是基于一些距离函数选择的最近的k个邻居。

在本文中，我们将重点介绍二元分类，为了防止平局k通常设置为奇数。与分类任务不同，在回归任务中，特征向量与实值标量而不是标签相关联，KNN是通过对响应变量均值或加权均值来进行预测。

惰性学习和非参数模型

惰性学习是 KNN 的标志。惰性学习器，也称为基于实例的学习器，这种学习器很少或根本不处理训练数据。与线性回归等积极学习的算法不同，KNN 不会估计在训练阶段概括训练数据的模型的参数。惰性学习有利有弊，训练一个积极学习的成本可能很高，但使用生成的模型进行预测的成本少。通过将系数乘以特征并添加偏置参数就可以预测简单的结果，计算成本低，预测速度快。但是一个惰性的学习者做出预测的成本是很高的，因为KNN 预测需要在计算测试实例和训练实例之间的距离，也就是要访问所有的训练数据。

参数模型使用固定数量的参数或系数来汇总数据。无论使用多少个训练实例，参数的数量都保持不变。非参数可能看起来用词不当，因为它并不意味着模型没有参数；相反，它意味着参数的数量随着训练数据的数量而变化。

当不熟悉响应变量和解释变量之间的关系时，非参数模型可能会很有用。KNN 就是这种非参数模型，如果实例彼此接近，则响应变量可能具有相似的值。当训练数据稀缺或已经知道这种关系时，带有假设的模型可能会比非参数模型有用。

使用 KNN 进行分类

我们使用一个简单的问题作为，我们需要根据一个人的身高和体重来预测他或她的性别的情况。这里有两个标签可以分配给响应变量，这就是为什么这个问题被称为二元分类。下表记录了九个训练实例：

KNN可以使用的特征没有数量限制，但不能可视化三个以上的特征（这是因为我们生活在3维空间，无法可视化更多维的数据）。通过创建散点图，可以使用matplotlib可视化数据:

从图中可以看到，男性往往比女性更高更重，这一点由x标记所示。我们的经验也与这个观察结果一致。要根据一个人的身高和体重来预测他/她的性别。需要定义距离测量指标：我们将测量两点之间的欧氏距离。在二维空间中，计算欧氏距离如下:

下一步是计算查询实例和所有训练实例之间的距离:

将k设置为3，我们将选择三个最近的实例来进行预测。在下面的脚本中，计算测试和训练实例之间的距离，并确定每个邻居的最常见性别:

在下面的图中，圆圈表示查询实例，放大的标记表示它最近的三个邻居:

有两个女性邻居和一个性男邻居。所以我们测试实例的结果是女性。使用scikit-learn实现KNN分类器，代码如下：

LabelBinarizer先将字符串转换为整数，fit方法创建了从标签字符串到整数的映射。输入标签使用transform方法进行转换。fit_transform同时调用fit和transform作。同时如果训练集和测试集是独立转换的，那么在训练集中男性可能映射为1，而在测试集中则映射为0。所以我们使用训练集的对象进行fit。然后使用KNeighborsClassifier进行预测。