K最近邻(k-Nearest Neighbor,KNN)分类算法是最简单的机器学习算法。(机器学习,算法本身不是最难的,最难的是:1、数学建模:把业务中的特性抽象成向量的过程;2、选取适合模型的数据样本。这两个事都不是简单的事。算法反而是比较简单的事。)
KNN算法的指导思想是“近朱者赤,近墨者黑”,由你的邻居来推断出你的类别。
本质上,KNN算法就是用距离来衡量样本之间的相似度
从训练集中找到和新数据最接近的k条记录,然后根据多数类来决定新数据类别。
算法涉及3个主要因素:
1) 训练数据集 2) 距离或相似度的计算衡量 3) k的大小
算法描述:
- 已知两类“先验”数据,分别是蓝方块和红三角,他们分布在一个二维空间中
- 有一个未知类别的数据(绿点),需要判断它是属于“蓝方块”还是“红三角”类
- 考察离绿点最近的3个(或k个)数据点的类别,占多数的类别即为绿点判定类别
计算步骤如下:
1)算距离:给定测试对象,计算它与训练集中的每个对象的距离 2)找邻居:圈定距离最近的k个训练对象,作为测试对象的近邻 3)做分类:根据这k个近邻归属的主要类别,来对测试对象分类
简单应用中,一般使用欧氏距离,但对于文本分类来说,使用余弦(cosine)来计算相似度就比欧式(Euclidean)距离更合适
该方法比较适用于样本容量比较大的类域的分类,而那些样本容量较小的类域采用这种算法比较容易产生误分。