K-近邻算法(kNN算法)

    技术2024-05-25  82

      最近在学习机器学习实战这本书,学的第一个算法就是K-近邻算法,话不多说,直接步入正题。

    一、K-近邻算法概述

    简单地说,K-近邻算法采用测量不同特征值之间的距离方法分类。

    优点:精度高、对异常值不敏感、无数据输入假定。

    缺点:计算复杂度高、空间复杂度高。

    适用数据范围:数值型和标称型。

    二、K-近邻算法的工作原理

    官方解释:存在一个样本数据集,也称作训练样本集,并且样本中每个数据都存在标签,即我们知道样本集中每一数据与所属分类的对应关系,输入没有标签的新数据后,将新数据的每个特征与样本集中的数据对应的特征进行比较,然后算法提取样本集中特征最相似的数据(最近邻)的分类标签。一般来说,我们只选择样本集中前k个最相似的数据,这就是k-近邻算法中k的出处,通常k是不大于20的整数,最后,选择k个最相似的数据中出现次数最多的分类,作为新数据的分类。

    我的理解:k-近邻算法就是根据“新数据的分类取决于它的邻居”进行的,比如邻居中大多数都是退伍军人,那么这个人也极有可能是退伍军人。而算法的目的就是先找出它的邻居,然后分析这几位邻居大多数的分类,极有可能就是它本省的分类。

    三、K-近邻算法的一般流程

    (1)收集数据:可以使用任何方法

    (2)准备数据:距离计算所需要的数值,最好是结构化的数据格式

    (3)分析数据:可以使用任何方法

    (4)训练算法:此步骤不适用于K-近邻算法

    (5)测试算法:计算错误率

    (6)使用算法:首先需要输入样本数据和结构化的输出结果,然后运行K-近邻算法判定输入数据分别属于哪个分类,最后应用对计算出的分类执行后续的处理 

    四、简单的操作实例

    1、准备:使用python导入数据

         创建KNN.py的python模块,并在python模块中添加以下代码

    from numpy import * import operator #导入运算符的包 def creatDataSet(): group = array([[1.0 , 1.1], [1.0 , 1.0], [0 , 0], [0 , 0.1]]) labels = ['A', 'A', 'B', 'B'] return group , labels

       我们可以简单的测试以下这段代码

    >>> group array([[1. , 1.1], [1. , 1. ], [0. , 0. ], [0. , 0.1]]) >>> labels ['A', 'A', 'B', 'B']

    以上是有四组数据每组数据有两个我们已知的属性或者特征值。上面的 group 矩阵每行包含一个不同的数据,我们可以把它想象为某个日志文件中不同的测量点或者入口。由于人类大脑的限制,我们通常只能可视化处理三维以下的事务。因此为了简单地实现数据可视化,对于每个数据点我们通常只使用两个特征。向量 labels 包含了每个数据点的标签信息, labels 包含的元素个数等于 group 矩阵行数。这里我们将数据点(1, 1.1)定义为类A,数据点(0, 0.1)定义为类B。带有4个数据点的简单例子如下图所示:

    2、实施kNN算法

    对未知类别属性的数据集中的每个点依次执行以下操作:

    (1)计算已知类别数据集中点与当前点之间的距离

    (2)按照距离递增次序排序

    (3)选取与当前点距离最小的k个点

    (4)确定前k个点所在类别出现的频率

    (5)返回前k个点出现频率最高的类别作为当前点的预测分类

    相关的实现代码如下(仍然填写在kNN.py的python模块中)

    def classify0(inX, dataSet, labels, k): dataSetSize = dataSet.shape[0] #表示第二维的长度,即行数 diffMat = tile(inX, (dataSetSize, 1)) - dataSet #tile表示铺成与训练集相同的维度 sqDiffMat = diffMat ** 2 sqDistances = sqDiffMat.sum(axis = 1) #sum(axis)指将矩阵的每一行相加 distances = sqDistances ** 0.5 sortedDistIndicies = distances.argsort() #将distances矩阵中的元素排序并返回相应的索引值 classCount = {} #创建一个字典来存储计数 for i in range(k): #K值是自己设置的 voteIlabel = labels[sortedDistIndicies[i]] #把按值大小顺序排列的欧氏距离索引list前k个对应的labels遍历出来 classCount[voteIlabel] = classCount.get(voteIlabel, 0) + 1 #计算前 k个出现的最多的次数 sortedClassCount = sorted(classCount.items(), key = operator.itemgetter(1), reverse = True) print(sortedClassCount) return sortedClassCount[0][0] #classCount.items()将classCount字典分解为元组列表 #operator.itemgetter(1)按照第二个元素次序对元组进行排序 #reverse = True是逆序,即按照从大到小的顺序进行排列 import kNN group, labels = kNN.creatDataSet() ank = kNN.classify0([0, 0], group, labels, 3) print('[0 , 0]所对应的类别为:', ank)

    运行上述代码可得以下结果:

    [('B', 2), ('A', 1)] [0 , 0]所对应的类别为: B

    从运行的结果可以看出[0 , 0]所对应的类别为‘B’类,大家可以改变[0 , 0]为其他值测试输出的结果

    3、代码解析

    (1)diffMat = tile(inX, (dataSetSize, 1)) - dataSet

             tile表示铺成与训练集相同的维度,举个栗子

    >>> A = array([1, 2, 3, 4]) >>> A array([1, 2, 3, 4]) >>> tile(A, (2, 1)) array([[1, 2, 3, 4], [1, 2, 3, 4]])

    (2)sqDistances = sqDiffMat.sum(axis = 1) 

             sum(axis = 1)指将矩阵的每一行相加,举个栗子

    >>> A = array([[1, 2, 3, 4], [0, 2, 4, 1]]) >>> A array([[1, 2, 3, 4], [0, 2, 4, 1]]) >>> A.sum(axis = 1) array([10, 7])

    (3)sortedDistIndicies = distances.argsort() 

            将distances矩阵中的元素排序并返回相应的索引值,举个栗子

    >>> A = array([-1, 4, 0, 1, 2]) >>> A.argsort() array([0, 2, 3, 4, 1], dtype=int64)

    (4)sortedClassCount = sorted(classCount.items(), key = operator.itemgetter(1), reverse = True)

             classCount.items()将classCount字典分解为元组列表          operator.itemgetter(1)按照第二个元素次序对元组进行排序          reverse = True是逆序,即按照从大到小的顺序进行排列          举个栗子

    >>> import operator >>> dict={} >>> dict['A']=2 >>> dict['B']=1 >>> dict['C']=3 >>> print dict {'A': 2, 'C': 3, 'B': 1} #创建了一个字典,名字为dict >>>Countlabels_max=sorted(dict.iteritems(),key=operator.itemgetter(1),reverse=True)#运行代码 >>> print Countlabels_max [('C', 3), ('A', 2), ('B', 1)] >>> Countlabels_max[0][0]#返回第0个tuple的第0个参数,也就是我最终的结果

    总结:kNN算法是个相对来说比较简单的算法,我也是初学者,如果有什么理解不到位的地方还请指点改正

     

    Processed: 0.010, SQL: 9