【放假第六天】KNN 文本分类 简单数字文字识别

#仅供自己学习笔记用
主要是前面的img2vector是新的知识点，也许基础知识会，能理解，但是自己写真的能卸出来吗
returnVect要先预设 然后往里面装 装的时候是分行读的，要怎么写？ int有注意到吗

import numpy as np
import os
from com.bjsxt.knn.KNNDateOnHand import classify
#就是封装好的包呗

def img2vector(filename):
    returnVect=np.zeros((1,1024))#原数据是32*32，而且注意是双重括号
    fr=open(filename)
    for line in range(32):
        lineStr=fr.readline()
        for j in range(32):
            returnVect[0,32*i+j]=int(lineStr[j])
    return returnVect


def IdentifImgClassText():
    hwLabel=[]
    trainingFileList=os.listdir('TrainData')  #https://www.runoob.com/python/os-listdir.html
    m=len(trainingFileList)
    trainingMat=np.zeros((m,1024))
    for i in range(m):
        fileNameStr=trainingFileList[i]
        fileStr=fileNameStr.split(".")[0]
        fileNumStr=int(fileStr.split("_")[0])
        hwLabes.append(classNumStr)
        trainingMat[i,:]=img2vector('TrainData/%s'%fileNameStr)


#然后也是又写了一遍，没有写循环呀，不是简便的就是好的呀

testFileList=os.listdir('TestData')
errorCount=0.0
mTest=len(testFileList)
for i in range(mTest):
    fileNameStr=trainingFileList[i]
    fileStr=fileNameStr.split(".")[0]
    fileNumStr=int(fileStr.split("_")[0])  #记得要化成整数型
    vectorUnderTest=img2vector('TestData/%s'%fileNumStr)
    classifierResult=classify(vectorUnderTest,trainingMat,hwLabels,3)
    print("识别出来的数字是： %d，真实的数字是：%d"(classifierResult,classNumStr))
    if(classifierResult!=classNumStr):
        errorCount+=1.0
    print("\n识别错误的个数%d"%errorCount)
    errorRate=errorCount/float(mTest)
    print("\n识别错误率%f"%erroRate)
    
    
if_name_=="_main_":
   #https://blog.csdn.net/Nancy50/article/details/90205367
    IdentifyImgClassTest