Algorithms/Statistical-Learning-Method_Code
1
0
Fork 0
mirror of https://gitee.com/TheAlgorithms/Statistical-Learning-Method_Code.git synced 2025-01-03 01:12:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update KNN.py

Browse Source
This commit is contained in:
Dod-o 2020-01-23 01:31:20 +08:00 committed by GitHub
parent d19687aba3
commit 5efeaa9b8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
KNN/KNN.py
View File

@ -120,7 +120,7 @@ def getClosest(trainDataMat, trainLabelMat, x, topK):
return labelList.index(max(labelList)) return labelList.index(max(labelList))
def test(trainDataArr, trainLabelArr, testDataArr, testLabelArr, topK): def model_test(trainDataArr, trainLabelArr, testDataArr, testLabelArr, topK):
''' '''
测试正确率 测试正确率
:param trainDataArr:训练集数据集 :param trainDataArr:训练集数据集
@ -166,7 +166,7 @@ if __name__ == "__main__":
#获取测试集 #获取测试集
testDataArr, testLabelArr = loadData('../Mnist/mnist_test.csv') testDataArr, testLabelArr = loadData('../Mnist/mnist_test.csv')
#计算测试集正确率 #计算测试集正确率
accur = test(trainDataArr, trainLabelArr, testDataArr, testLabelArr, 25) accur = model_test(trainDataArr, trainLabelArr, testDataArr, testLabelArr, 25)
#打印正确率 #打印正确率
print('accur is:%d'%(accur * 100), '%') print('accur is:%d'%(accur * 100), '%')