热线电话：13121318867

登录
首页精彩阅读机器学习python实战之手写数字识别
机器学习python实战之手写数字识别
2018-05-21
收藏

机器学习python实战之手写数字识别

看了上一篇内容之后，相信对K近邻算法有了一个清晰的认识，今天的内容——手写数字识别是对上一篇内容的延续，这里也是为了自己能更熟练的掌握k-NN算法。

我们有大约2000个训练样本和1000个左右测试样本，训练样本所在的文件夹是trainingDigits，测试样本所在的文件夹是testDigits。文本文件中是0~9的数字，但是是用二值图表示出来的，如图。我们要做的就是使用训练样本训练模型，并用测试样本来检测模型的性能。

首先，我们需要将文本文件中的内容转化为向量，因为图片大小是32*32，所以我们可以将其转化为1*1024的向量。具体代码实现如下：

defimg2vector(filename):
  imgVec=zeros((1,1024))
  file=open(filename)
  foriinrange(32):
    lines=file.readline()
    forjinrange(32):
      imgVec[0,32*i+j]=lines[j]
  returnimgVec
实现了图片到向量的转化之后，我们就可以对测试文件中的内容进行识别了。这里的识别我们可以使用上一篇中的自定义函数classify0，这个函数的第一个参数是测试向量，第二个参数是训练数据集，第三个参数是训练集的标签。所以，我们首先需要将训练数据集转化为（1934*1024）的矩阵，1934这里是训练集的组数即trainingDigits目录下的文件数，其对应的标签转化为（1*1934）的向量。之后要编写的代码就是对测试数据集中的每个文本文件进行识别，也就是需要将每个文件都转化成一个（1*1024）的向量，再传入classify0函数的第一个形参。整体代码如下：

defhandWriteNumClassTest():
  NumLabels=[]
  TrainingDirfile=listdir(r'D:\ipython\num_recognize\trainingDigits')#文件目录
  L=len(TrainingDirfile) #该目录中有多少文件
  TrainMat=zeros((L,1024))
  foriinrange(L):
    file_n=TrainingDirfile[i]
    fileName=file_n.split('.')[0]
    ClassName=int(file_n.split('_')[0])
    NumLabels.append(ClassName)
    TrainMat[i,:]=img2vector(r'D:\ipython\num_recognize\trainingDigits\%s'%file_n)
  TestfileDir=listdir(r'D:\ipython\num_recognize\testDigits')
  error_cnt=0.0
  M=len(TestfileDir)
  forjinrange(M):
    Testfile=TestfileDir[j]
    TestfileName=Testfile.split('.')[0]
    TestClassName=int(Testfile.split('_')[0])
    TestVector=img2vector(r'D:\ipython\num_recognize\testDigits\%s'%Testfile)
    result=classify0(TestVector,TrainMat,NumLabels,3)
    print('the result is %d,the real answer is %d\n'%(result,TestClassName))
    ifresult!=TestClassName:
      error_cnt+=1
  print('the total num of errors is %f\n'%error_cnt)
  print('the error rate is %f\n'%(error_cnt/float(M)))

这里需要首先导入listdir方法，from os import listdir，它可以列出给定目录的文件名。对于测试的每个文件，如果识别的分类结果跟真实结果不一样，则错误数+1，最终用错误数/测试总数 来表示该模型的性能。下面给出结果

这里测试的总共946个项目中，一共有10个出现了错误，出错率为1%，这个性能还是可以接受的。有了上一篇内容的理解，这篇就简单多了吧！


CDA数据分析师考试相关入口一览（建议收藏）：

▷ 想报名CDA认证考试，点击>>>   “CDA报名” 了解CDA考试详情；

▷ 想加入CDA考试题库，点击>>> “CDA题库” 了解CDA考试详情；

▷ 想学习CDA考试教材，点击>>> “CDA教材” 了解CDA考试详情；

▷ 想查询CDA考试成绩，点击>>> “CDA成绩” 了解CDA考试详情；

▷ 想了解CDA考试含金量，点击>>> “CDA含金量” 了解CDA考试详情；

▷ 想获取CDA考试时间/费用/条件/大纲/通过率，点击 >>>“CDA考试官网” 了解CDA考试详情；

python 机器学习

数据分析咨询请扫描二维码

上一篇大数据热门词汇汇总
下一篇用R语言进行关联分析
CDA考试动态
CDA报考指南
热门栏目
最新资讯
更多
Copyright © 2015-present 北京国富如荷网络科技有限公司 All Rights Reserved.
京ICP备11001960号-9   京公网安备 11010802034615号   经营许可证编号：京B2-20210330
客服在线
立即咨询