hwLabels.append(classnumCut(fileNameStr))
trainingMat[i,:]=img2vecor(file+'/%s' % fileNameStr)
return hwLabels,trainingMat
def handwritingTest(file):
hwLabels,trainingMat = trainingDataSet('/root/python_test/data/data/trainingDigits') #构建训练集
testFileList = listdir(file) #获取测试集
errorCount = 0.0 #错误数
mTest = len(testFileList) #测试集总样本数
t1 = time.time()
for i in range(mTest):
fileNameStr = testFileList
classNumStr = classnumCut(fileNameStr)
vectorUnderTest=img2vecor(file+'/%s' % fileNameStr)
classifierResult =classify(vectorUnderTest,trainingMat,hwLabels, 3)
print("the classifier came back with: %d, the real answer is: %d" % (classifierResult, classNumStr))
if (classifierResult != classNumStr):
errorCount += 1.0
print ("\nthe total number of tests is: %d" % mTest) #输出测试总样本数
print ("the total number of errors is: %d" % errorCount) #输出测试错误样本数
print("the total error rate is: %f" % (errorCount/float(mTest))) #输出错误率
t2 = time.time()
print("Cost time: %.2fmin, %.4fs."%((t2-t1)//60,(t2-t1)%60) ) #测试耗时
if __name__ == "__main__":
handwritingTest('/root/python_test/data/data/testDigits')
---------------------
作者:wx_411180165
来源:CSDN
原文:https://blog.csdn.net/qq_24726509/article/details/84923274
版权声明:本文为博主原创文章,转载请附上博文链接!