k近邻算法

k近邻算法(kNN)

        本博客来源于CSDN:http://blog.csdn.net/niuwei22007/article/details/49703719

              本博客源代码下载地址:CSDN免费下载GitHub下载地址 均带有详细注释和测试数据

        今天学习了《机器学习实战》这本书介绍的第一个机器学习算法—k近邻算法。书中介绍它对于分类非常有效,比如书中的例子是对电影的题材进行分类。

一、算法原理

        算法原理是什么?允许我不严谨的说一下:首先有一堆有标签的样本,比如有一堆各种各样的鸟(样本集),我知道各种鸟的不同外貌(特征),比如羽毛颜色有无脚蹼身体重量身体长度以及最重要的它属于哪一种鸟类别/标签);然后给我一只不是这堆鸟中的一只鸟(测试样本),让我观察了它的羽毛颜色等后,让我说出它属于哪一种鸟?我的做法是:遍历之前的一堆鸟,分别比较每一只鸟的羽毛颜色、身体重量等特征与给定鸟的相应特征,并给出这两只鸟的相似度。最终,从那一堆鸟中找出相似度最大的前k,然后统计这k只鸟的分类,最后把分类数量最多的那只鸟的类别作为给定的类别。虽然结果不一定准确,但是是有理论支持的,那就是概率论,哈哈。

        下面来看一下书上对这个算法的原理介绍:存在一个训练样本集,并且每个样本都存在标签(有监督学习)。输入没有标签的新样本数据后,将新数据的每个特征与样本集中数据对应的特征进行比较,然后算法提取出与样本集中特征最相似的数据(最近邻)的分类标签。一般来说,我们只选择样本数据集中前k个最相似的数据,这就是k-近邻算法中k的出处,而且k通常不大于20。最后选择k个最相似数据中出现次数最多的分类,作为新数据的分类。

二、如何解决问题

        没接触过的同学应该能懂了吧。书中的举例是对电影的题材进行分类:爱情片or动作片。依据电影中打斗镜头和接吻镜头的数量。下面来看一下如何用kNN来解决这个问题。

        

        要解给定一部电影,判断其属于哪一种电影这个问题,就需要知道这个未知电影存在多少个打斗镜头和接吻镜头,上图所示,问号位置所代表的两种镜头次数分别是多少?

        下面我们来看一下图中电影的特征值,如下表:


        相信看过数据以后,即使不知道未知电影(?)属于哪一种类型,但是可以通过某个计算方法计算出来。

  • 第一步:首先计算未知电影与已知电影的相似度(抽象距离--相似度越小,距离越远)。具体如何计算暂且不考虑。下面看一下相似度列表:
  • 第二步:再将相似度列表排序,选出前k个最相似的样本。此处我们假设k=3,将上表中的相似度进行排序后前3分别是:He’s Not Really into DudesBeautiful WomanCalifornia Man
  • 第三步:统计最相似样本的分类。此处很容易知道这3个样本均为爱情片。
  • 第四步:将分类最多的类别作为未知电影的分类。那么我们就得出结论,未知电影属于爱情片。

        下面贴一下书上总结的k近邻算法的一般流程:


三、用python实现kNN算法之简单分类

  1. #coding=UTF8  
  2. from numpy import *  
  3. import operator  
  4.   
  5. def createDataSet():  
  6.     """ 
  7.     函数作用:构建一组训练数据(训练样本),共4个样本 
  8.     同时给出了这4个样本的标签,及labels 
  9.     """  
  10.     group = array([  
  11.         [1.01.1],  
  12.         [1.01.0],  
  13.         [0. , 0. ],  
  14.         [0. , 0.1]  
  15.     ])  
  16.     labels = ['A''A''B''B']  
  17.     return group, labels  
  18.   
  19. def classify0(inX, dataset, labels, k):  
  20.     """ 
  21.     inX 是输入的测试样本,是一个[x, y]样式的 
  22.     dataset 是训练样本集 
  23.     labels 是训练样本标签 
  24.     k 是top k最相近的 
  25.     """  
  26.     # shape返回矩阵的[行数,列数],  
  27.     # 那么shape[0]获取数据集的行数,  
  28.     # 行数就是样本的数量  
  29.     dataSetSize = dataset.shape[0]   
  30.       
  31.     """ 
  32.     下面的求距离过程就是按照欧氏距离的公式计算的。 
  33.     即 根号(x^2+y^2) 
  34.     """  
  35.     # tile属于numpy模块下边的函数  
  36.     # tile(A, reps)返回一个shape=reps的矩阵,矩阵的每个元素是A  
  37.     # 比如 A=[0,1,2] 那么,tile(A, 2)= [0, 1, 2, 0, 1, 2]  
  38.     # tile(A,(2,2)) = [[0, 1, 2, 0, 1, 2],  
  39.     #                  [0, 1, 2, 0, 1, 2]]  
  40.     # tile(A,(2,1,2)) = [[[0, 1, 2, 0, 1, 2]],  
  41.     #                    [[0, 1, 2, 0, 1, 2]]]   
  42.     # 上边那个结果的分开理解就是:  
  43.     # 最外层是2个元素,即最外边的[]中包含2个元素,类似于[C,D],而此处的C=D,因为是复制出来的  
  44.     # 然后C包含1个元素,即C=[E],同理D=[E]  
  45.     # 最后E包含2个元素,即E=[F,G],此处F=G,因为是复制出来的  
  46.     # F就是A了,基础元素  
  47.     # 综合起来就是(2,1,2)= [C, C] = [[E], [E]] = [[[F, F]], [[F, F]]] = [[[A, A]], [[A, A]]]  
  48.     # 这个地方就是为了把输入的测试样本扩展为和dataset的shape一样,然后就可以直接做矩阵减法了。  
  49.     # 比如,dataset有4个样本,就是4*2的矩阵,输入测试样本肯定是一个了,就是1*2,为了计算输入样本与训练样本的距离  
  50.     # 那么,需要对这个数据进行作差。这是一次比较,因为训练样本有n个,那么就要进行n次比较;  
  51.     # 为了方便计算,把输入样本复制n次,然后直接与训练样本作矩阵差运算,就可以一次性比较了n个样本。  
  52.     # 比如inX = [0,1],dataset就用函数返回的结果,那么  
  53.     # tile(inX, (4,1))= [[ 0.0, 1.0],  
  54.     #                    [ 0.0, 1.0],  
  55.     #                    [ 0.0, 1.0],  
  56.     #                    [ 0.0, 1.0]]  
  57.     # 作差之后  
  58.     # diffMat = [[-1.0,-0.1],  
  59.     #            [-1.0, 0.0],  
  60.     #            [ 0.0, 1.0],  
  61.     #            [ 0.0, 0.9]]  
  62.     diffMat = tile(inX, (dataSetSize, 1)) - dataset  
  63.       
  64.     # diffMat就是输入样本与每个训练样本的差值,然后对其每个x和y的差值进行平方运算。  
  65.     # diffMat是一个矩阵,矩阵**2表示对矩阵中的每个元素进行**2操作,即平方。  
  66.     # sqDiffMat = [[1.0, 0.01],  
  67.     #              [1.0, 0.0 ],  
  68.     #              [0.0, 1.0 ],  
  69.     #              [0.0, 0.81]]  
  70.     sqDiffMat = diffMat ** 2  
  71.       
  72.     # axis=1表示按照横轴,sum表示累加,即按照行进行累加。  
  73.     # sqDistance = [[1.01],  
  74.     #               [1.0 ],  
  75.     #               [1.0 ],  
  76.     #               [0.81]]  
  77.     sqDistance = sqDiffMat.sum(axis=1)  
  78.       
  79.     # 对平方和进行开根号  
  80.     distance = sqDistance ** 0.5  
  81.       
  82.     # 按照升序进行快速排序,返回的是原数组的下标。  
  83.     # 比如,x = [30, 10, 20, 40]  
  84.     # 升序排序后应该是[10,20,30,40],他们的原下标是[1,2,0,3]  
  85.     # 那么,numpy.argsort(x) = [1, 2, 0, 3]  
  86.     sortedDistIndicies = distance.argsort()  
  87.       
  88.     # 存放最终的分类结果及相应的结果投票数  
  89.     classCount = {}  
  90.       
  91.     # 投票过程,就是统计前k个最近的样本所属类别包含的样本个数  
  92.     for i in range(k):  
  93.         # index = sortedDistIndicies[i]是第i个最相近的样本下标  
  94.         # voteIlabel = labels[index]是样本index对应的分类结果('A' or 'B')  
  95.         voteIlabel = labels[sortedDistIndicies[i]]  
  96.         # classCount.get(voteIlabel, 0)返回voteIlabel的值,如果不存在,则返回0  
  97.         # 然后将票数增1  
  98.         classCount[voteIlabel] = classCount.get(voteIlabel, 0) + 1  
  99.       
  100.     # 把分类结果进行排序,然后返回得票数最多的分类结果  
  101.     sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True)  
  102.     return sortedClassCount[0][0]  
  103.   
  104. if __name__== "__main__":  
  105.     # 导入数据  
  106.     dataset, labels = createDataSet()  
  107.     inX = [0.10.1]  
  108.     # 简单分类  
  109.     className = classify0(inX, dataset, labels, 3)  
  110.     print 'the class of test sample is %s' %className  

四、用python实现kNN算法之改进分类


  1. #coding=UTF8  
  2. from numpy import *  
  3. import operator  
  4. from os import listdir  
  5.   
  6. def classify0(inX, dataset, labels, k):  
  7.     """ 
  8.     inX 是输入的测试样本,是一个[x, y]样式的 
  9.     dataset 是训练样本集 
  10.     labels 是训练样本标签 
  11.     k 是top k最相近的 
  12.     """  
  13.     # shape返回矩阵的[行数,列数],  
  14.     # 那么shape[0]获取数据集的行数,  
  15.     # 行数就是样本的数量  
  16.     dataSetSize = dataset.shape[0]   
  17.       
  18.     """ 
  19.     下面的求距离过程就是按照欧氏距离的公式计算的。 
  20.     即 根号(x^2+y^2) 
  21.     """  
  22.     # tile属于numpy模块下边的函数  
  23.     # tile(A, reps)返回一个shape=reps的矩阵,矩阵的每个元素是A  
  24.     # 比如 A=[0,1,2] 那么,tile(A, 2)= [0, 1, 2, 0, 1, 2]  
  25.     # tile(A,(2,2)) = [[0, 1, 2, 0, 1, 2],  
  26.     #                  [0, 1, 2, 0, 1, 2]]  
  27.     # tile(A,(2,1,2)) = [[[0, 1, 2, 0, 1, 2]],  
  28.     #                    [[0, 1, 2, 0, 1, 2]]]   
  29.     # 上边那个结果的分开理解就是:  
  30.     # 最外层是2个元素,即最外边的[]中包含2个元素,类似于[C,D],而此处的C=D,因为是复制出来的  
  31.     # 然后C包含1个元素,即C=[E],同理D=[E]  
  32.     # 最后E包含2个元素,即E=[F,G],此处F=G,因为是复制出来的  
  33.     # F就是A了,基础元素  
  34.     # 综合起来就是(2,1,2)= [C, C] = [[E], [E]] = [[[F, F]], [[F, F]]] = [[[A, A]], [[A, A]]]  
  35.     # 这个地方就是为了把输入的测试样本扩展为和dataset的shape一样,然后就可以直接做矩阵减法了。  
  36.     # 比如,dataset有4个样本,就是4*2的矩阵,输入测试样本肯定是一个了,就是1*2,为了计算输入样本与训练样本的距离  
  37.     # 那么,需要对这个数据进行作差。这是一次比较,因为训练样本有n个,那么就要进行n次比较;  
  38.     # 为了方便计算,把输入样本复制n次,然后直接与训练样本作矩阵差运算,就可以一次性比较了n个样本。  
  39.     # 比如inX = [0,1],dataset就用函数返回的结果,那么  
  40.     # tile(inX, (4,1))= [[ 0.0, 1.0],  
  41.     #                    [ 0.0, 1.0],  
  42.     #                    [ 0.0, 1.0],  
  43.     #                    [ 0.0, 1.0]]  
  44.     # 作差之后  
  45.     # diffMat = [[-1.0,-0.1],  
  46.     #            [-1.0, 0.0],  
  47.     #            [ 0.0, 1.0],  
  48.     #            [ 0.0, 0.9]]  
  49.     diffMat = tile(inX, (dataSetSize, 1)) - dataset  
  50.       
  51.     # diffMat就是输入样本与每个训练样本的差值,然后对其每个x和y的差值进行平方运算。  
  52.     # diffMat是一个矩阵,矩阵**2表示对矩阵中的每个元素进行**2操作,即平方。  
  53.     # sqDiffMat = [[1.0, 0.01],  
  54.     #              [1.0, 0.0 ],  
  55.     #              [0.0, 1.0 ],  
  56.     #              [0.0, 0.81]]  
  57.     sqDiffMat = diffMat ** 2  
  58.       
  59.     # axis=1表示按照横轴,sum表示累加,即按照行进行累加。  
  60.     # sqDistance = [[1.01],  
  61.     #               [1.0 ],  
  62.     #               [1.0 ],  
  63.     #               [0.81]]  
  64.     sqDistance = sqDiffMat.sum(axis=1)  
  65.       
  66.     # 对平方和进行开根号  
  67.     distance = sqDistance ** 0.5  
  68.       
  69.     # 按照升序进行快速排序,返回的是原数组的下标。  
  70.     # 比如,x = [30, 10, 20, 40]  
  71.     # 升序排序后应该是[10,20,30,40],他们的原下标是[1,2,0,3]  
  72.     # 那么,numpy.argsort(x) = [1, 2, 0, 3]  
  73.     sortedDistIndicies = distance.argsort()  
  74.       
  75.     # 存放最终的分类结果及相应的结果投票数  
  76.     classCount = {}  
  77.       
  78.     # 投票过程,就是统计前k个最近的样本所属类别包含的样本个数  
  79.     for i in range(k):  
  80.         # index = sortedDistIndicies[i]是第i个最相近的样本下标  
  81.         # voteIlabel = labels[index]是样本index对应的分类结果('A' or 'B')  
  82.         voteIlabel = labels[sortedDistIndicies[i]]  
  83.         # classCount.get(voteIlabel, 0)返回voteIlabel的值,如果不存在,则返回0  
  84.         # 然后将票数增1  
  85.         classCount[voteIlabel] = classCount.get(voteIlabel, 0) + 1  
  86.       
  87.     # 把分类结果进行排序,然后返回得票数最多的分类结果  
  88.     sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True)  
  89.     return sortedClassCount[0][0]  
  90.       
  91. def file2matrix(filename):  
  92.     """ 
  93.     从文件中读入训练数据,并存储为矩阵 
  94.     """  
  95.     fr = open(filename)  
  96.     numberOfLines = len(fr.readlines())         #获取 n=样本的行数  
  97.     returnMat = zeros((numberOfLines,3))        #创建一个2维矩阵用于存放训练样本数据,一共有n行,每一行存放3个数据  
  98.     classLabelVector = []                       #创建一个1维数组用于存放训练样本标签。    
  99.     fr = open(filename)  
  100.     index = 0  
  101.     for line in fr.readlines():  
  102.         # 把回车符号给去掉  
  103.         line = line.strip()      
  104.         # 把每一行数据用\t分割  
  105.         listFromLine = line.split('\t')  
  106.         # 把分割好的数据放至数据集,其中index是该样本数据的下标,就是放到第几行  
  107.         returnMat[index,:] = listFromLine[0:3]  
  108.         # 把该样本对应的标签放至标签集,顺序与样本集对应。  
  109.         classLabelVector.append(int(listFromLine[-1]))  
  110.         index += 1  
  111.     return returnMat,classLabelVector  
  112.       
  113. def autoNorm(dataSet):  
  114.     """ 
  115.     训练数据归一化 
  116.     """  
  117.     # 获取数据集中每一列的最小数值  
  118.     # 以createDataSet()中的数据为例,group.min(0)=[0,0]  
  119.     minVals = dataSet.min(0)   
  120.     # 获取数据集中每一列的最大数值  
  121.     # group.max(0)=[1, 1.1]  
  122.     maxVals = dataSet.max(0)   
  123.     # 最大值与最小的差值  
  124.     ranges = maxVals - minVals  
  125.     # 创建一个与dataSet同shape的全0矩阵,用于存放归一化后的数据  
  126.     normDataSet = zeros(shape(dataSet))  
  127.     m = dataSet.shape[0]  
  128.     # 把最小值扩充为与dataSet同shape,然后作差,具体tile请翻看 第三节 代码中的tile  
  129.     normDataSet = dataSet - tile(minVals, (m,1))  
  130.     # 把最大最小差值扩充为dataSet同shape,然后作商,是指对应元素进行除法运算,而不是矩阵除法。  
  131.     # 矩阵除法在numpy中要用linalg.solve(A,B)  
  132.     normDataSet = normDataSet/tile(ranges, (m,1))  
  133.     return normDataSet, ranges, minVals  
  134.      
  135. def datingClassTest():  
  136.     # 将数据集中10%的数据留作测试用,其余的90%用于训练  
  137.     hoRatio = 0.10  
  138.     datingDataMat,datingLabels = file2matrix('datingTestSet2.txt')       #load data setfrom file  
  139.     normMat, ranges, minVals = autoNorm(datingDataMat)  
  140.     m = normMat.shape[0]  
  141.     numTestVecs = int(m*hoRatio)  
  142.     errorCount = 0.0  
  143.     for i in range(numTestVecs):  
  144.         classifierResult = classify0(normMat[i,:],normMat[numTestVecs:m,:],datingLabels[numTestVecs:m],3)  
  145.         print "the classifier came back with: %d, the real answer is: %d, result is :%s" % (classifierResult, datingLabels[i],classifierResult==datingLabels[i])  
  146.         if (classifierResult != datingLabels[i]): errorCount += 1.0  
  147.     print "the total error rate is: %f" % (errorCount/float(numTestVecs))  
  148.     print errorCount  
  149.   
  150. if __name__=="__main__"  
  151.     datingClassTest()  

        改进一:数值归一化

        第三节就是按照第二节中描述实现的算法。但是这只是最简单的分类,当遇到特征值相差很大的时候,分类效果就不好了。比如,原来的dataSet中有一个这样的样本:[1, 10000]。同时输入样本仍然为第三节中测试的[0.1, 0.1]。所以很明显,当求这两个样本之间的距离的时候就出问题,由于那个10000太大,对最终的结果影响最大,其他的0.1和1可以忽略不计了。因此就用到了数值归一化。如果还有不太清楚地,百度一下。详细实现看代码,有详细注释。

        改进二:从文件中读入数据

        要做一个稍微复杂一点的测试,就需要多准备点样本数据,而在第三节中,训练样本只有4个,而且是固定生成的。在新的算法里边加入了可以从文件中读取训练数据。代码有详细注释。

五、用python实现kNN算法之手写数字识别

        识别手写数字,需要先把手写数字图片转换为相应的01矩阵格式,这里提供已经经过转换的数据文件。trainingDigits是2000个训练样本,testDigits是900个测试样本。

  1. #coding=UTF8  
  2. from numpy import *  
  3. import operator  
  4. from os import listdir  
  5.   
  6. def classify0(inX, dataset, labels, k):  
  7.     """  
  8.     inX 是输入的测试样本,是一个[x, y]样式的  
  9.     dataset 是训练样本集  
  10.     labels 是训练样本标签  
  11.     k 是top k最相近的  
  12.     """  
  13.     # shape返回矩阵的[行数,列数],  
  14.     # 那么shape[0]获取数据集的行数,  
  15.     # 行数就是样本的数量  
  16.     dataSetSize = dataset.shape[0]   
  17.       
  18.     # tile属于numpy模块下边的函数  
  19.     # tile(A, reps)返回一个shape=reps的矩阵,矩阵的每个元素是A  
  20.     # 比如 A=[0,1,2] 那么,tile(A, 2)= [012012]  
  21.     # tile(A,(2,2)) = [[012012],  
  22.     #                  [012012]]  
  23.     # tile(A,(2,1,2)) = [[[012012]],  
  24.     #                    [[012012]]]   
  25.     # 上边那个结果的分开理解就是:  
  26.     # 最外层是2个元素,即最外边的[]中包含2个元素,类似于[C,D],而此处的C=D,因为是复制出来的  
  27.     # 然后C包含1个元素,即C=[E],同理D=[E]  
  28.     # 最后E包含2个元素,即E=[F,G],此处F=G,因为是复制出来的  
  29.     # F就是A了,基础元素  
  30.     # 综合起来就是(2,1,2)= [C, C] = [[E], [E]] = [[[F, F]], [[F, F]]] = [[[A, A]], [[A, A]]]  
  31.     # 这个地方就是为了把输入的测试样本扩展为和dataset的shape一样,然后就可以直接做矩阵减法了。  
  32.     # 比如,dataset有4个样本,就是4*2的矩阵,输入测试样本肯定是一个了,就是1*2,为了计算输入样本与训练样本的距离  
  33.     # 那么,需要对这个数据进行作差。这是一次比较,因为训练样本有n个,那么就要进行n次比较;  
  34.     # 为了方便计算,把输入样本复制n次,然后直接与训练样本作矩阵差运算,就可以一次性比较了n个样本。  
  35.     # 比如inX = [0,1],dataset就用函数返回的结果,那么  
  36.     # tile(inX, (4,1))= [[ 0.01.0],  
  37.     #                    [ 0.01.0],  
  38.     #                    [ 0.01.0],  
  39.     #                    [ 0.01.0]]  
  40.     # 作差之后  
  41.     # diffMat = [[-1.0,-0.1],  
  42.     #            [-1.00.0],  
  43.     #            [ 0.01.0],  
  44.     #            [ 0.00.9]]  
  45.     diffMat = tile(inX, (dataSetSize, 1)) - dataset  
  46.       
  47.     # diffMat就是输入样本与每个训练样本的差值,然后对其每个x和y的差值进行平方运算。  
  48.     # diffMat是一个矩阵,矩阵**2表示对矩阵中的每个元素进行**2操作,即平方。  
  49.     # sqDiffMat = [[1.00.01],  
  50.     #              [1.00.0 ],  
  51.     #              [0.01.0 ],  
  52.     #              [0.00.81]]  
  53.     sqDiffMat = diffMat ** 2  
  54.       
  55.     # axis=1表示按照横轴,sum表示累加,即按照行进行累加。  
  56.     # sqDistance = [[1.01],  
  57.     #               [1.0 ],  
  58.     #               [1.0 ],  
  59.     #               [0.81]]  
  60.     sqDistance = sqDiffMat.sum(axis=1)  
  61.       
  62.     # 对平方和进行开根号  
  63.     distance = sqDistance ** 0.5  
  64.       
  65.     # 按照升序进行快速排序,返回的是原数组的下标。  
  66.     # 比如,x = [30102040]  
  67.     # 升序排序后应该是[10,20,30,40],他们的原下标是[1,2,0,3]  
  68.     # 那么,numpy.argsort(x) = [1203]  
  69.     sortedDistIndicies = distance.argsort()  
  70.       
  71.     # 存放最终的分类结果及相应的结果投票数  
  72.     classCount = {}  
  73.       
  74.     # 投票过程,就是统计前k个最近的样本所属类别包含的样本个数  
  75.     for i in range(k):  
  76.         # index = sortedDistIndicies[i]是第i个最相近的样本下标  
  77.         # voteIlabel = labels[index]是样本index对应的分类结果('A' or 'B')  
  78.         voteIlabel = labels[sortedDistIndicies[i]]  
  79.         # classCount.get(voteIlabel, 0)返回voteIlabel的值,如果不存在,则返回0  
  80.         # 然后将票数增1  
  81.         classCount[voteIlabel] = classCount.get(voteIlabel, 0) + 1  
  82.       
  83.     # 把分类结果进行排序,然后返回得票数最多的分类结果  
  84.     sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True)  
  85.     return sortedClassCount[0][0]  
  86.       
  87. def file2matrix(filename):  
  88.     """  
  89.     从文件中读入训练数据,并存储为矩阵  
  90.     """  
  91.     fr = open(filename)  
  92.     numberOfLines = len(fr.readlines())         #获取 n=样本的行数  
  93.     returnMat = zeros((numberOfLines,3))        #创建一个2维矩阵用于存放训练样本数据,一共有n行,每一行存放3个数据  
  94.     classLabelVector = []                       #创建一个1维数组用于存放训练样本标签。    
  95.     fr = open(filename)  
  96.     index = 0  
  97.     for line in fr.readlines():  
  98.         # 把回车符号给去掉  
  99.         line = line.strip()      
  100.         # 把每一行数据用\t分割  
  101.         listFromLine = line.split('\t')  
  102.         # 把分割好的数据放至数据集,其中index是该样本数据的下标,就是放到第几行  
  103.         returnMat[index,:] = listFromLine[0:3]  
  104.         # 把该样本对应的标签放至标签集,顺序与样本集对应。  
  105.         classLabelVector.append(int(listFromLine[-1]))  
  106.         index += 1  
  107.     return returnMat,classLabelVector  
  108.       
  109. def autoNorm(dataSet):  
  110.     """  
  111.     训练数据归一化  
  112.     """  
  113.     # 获取数据集中每一列的最小数值  
  114.     # 以createDataSet()中的数据为例,group.min(0)=[0,0]  
  115.     minVals = dataSet.min(0)   
  116.     # 获取数据集中每一列的最大数值  
  117.     # group.max(0)=[11.1]  
  118.     maxVals = dataSet.max(0)   
  119.     # 最大值与最小的差值  
  120.     ranges = maxVals - minVals  
  121.     # 创建一个与dataSet同shape的全0矩阵,用于存放归一化后的数据  
  122.     normDataSet = zeros(shape(dataSet))  
  123.     m = dataSet.shape[0]  
  124.     # 把最小值扩充为与dataSet同shape,然后作差,具体tile请翻看 第三节 代码中的tile  
  125.     normDataSet = dataSet - tile(minVals, (m,1))  
  126.     # 把最大最小差值扩充为dataSet同shape,然后作商,是指对应元素进行除法运算,而不是矩阵除法。  
  127.     # 矩阵除法在numpy中要用linalg.solve(A,B)  
  128.     normDataSet = normDataSet/tile(ranges, (m,1))  
  129.     return normDataSet, ranges, minVals  
  130.      
  131. def img2vector(filename):  
  132.     """  
  133.     将图片数据转换为01矩阵。  
  134.     每张图片是32*32像素,也就是一共1024个字节。  
  135.     因此转换的时候,每行表示一个样本,每个样本含1024个字节。  
  136.     """  
  137.     # 每个样本数据是1024=32*32个字节  
  138.     returnVect = zeros((1,1024))  
  139.     fr = open(filename)  
  140.     # 循环读取32行,32列。  
  141.     for i in range(32):  
  142.         lineStr = fr.readline()  
  143.         for j in range(32):  
  144.             returnVect[0,32*i+j] = int(lineStr[j])  
  145.     return returnVect  
  146.   
  147. def handwritingClassTest():  
  148.     hwLabels = []  
  149.     # 加载训练数据  
  150.     trainingFileList = listdir('trainingDigits')             
  151.     m = len(trainingFileList)  
  152.     trainingMat = zeros((m,1024))  
  153.     for i in range(m):  
  154.         # 从文件名中解析出当前图像的标签,也就是数字是几  
  155.         fileNameStr = trainingFileList[i]  
  156.         fileStr = fileNameStr.split('.')[0]     #take off .txt  
  157.         classNumStr = int(fileStr.split('_')[0])  
  158.         hwLabels.append(classNumStr)  
  159.         trainingMat[i,:] = img2vector('trainingDigits/%s' % fileNameStr)  
  160.     # 加载测试数据  
  161.     testFileList = listdir('testDigits')        #iterate through the test set  
  162.     errorCount = 0.0  
  163.     mTest = len(testFileList)  
  164.     for i in range(mTest):  
  165.         fileNameStr = testFileList[i]  
  166.         fileStr = fileNameStr.split('.')[0]     #take off .txt  
  167.         classNumStr = int(fileStr.split('_')[0])  
  168.         vectorUnderTest = img2vector('testDigits/%s' % fileNameStr)  
  169.         classifierResult = classify0(vectorUnderTest, trainingMat, hwLabels, 3)  
  170.         print "the classifier came back with: %d, the real answer is: %d, The predict result is: %s" % (classifierResult, classNumStr, classifierResult==classNumStr)  
  171.         if (classifierResult != classNumStr): errorCount += 1.0  
  172.     print "\nthe total number of errors is: %d / %d" %(errorCount, mTest)  
  173.     print "\nthe total error rate is: %f" % (errorCount/float(mTest))  
  174.   
  175. if __name__== "__main__":  
  176.     handwritingClassTest()      

六、小结

        引用原话:k近邻算法是分类数据最简单最有效的算法。k紧邻算法必须保存全部数据集,如果训练数据集很大,必须使用大量的存储空间。此外,由于必须对数据集中的每个数据计算距离值,实际使用时可能非常耗时。(这些是很容易理解的)

        k近邻算法的另外一个缺陷是它无法给出任何数据的基础结构信息,因此我们也无法知晓平均实例样本和典型实例样本具有什么特征。(就是说k近邻算法只能是单纯的比较样本特征值之间的大小关系,给出样本之间的欧氏距离,并不能给出这些特征数据的内在含义。还是比较抽象!用一句不太恰当的描述是:k近邻算法只是知道两个样本之间的有区别,长得像不像,机械的区分样本之间的差别,没有类似于人的理解能力。)

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值