基于python的手写数字识别knn_用sklearn中的KNN实现Kaggle手写数字识别

import csv

from sklearn import neighbors

#导入训练数据和测试数据

def loadData(filename1,filename2,trainDataSet,trainTargetSet,testDataSet):

with open(filename1,'r') as csvfile1:

lines1 = csv.reader(csvfile1)

dataSet = list(lines1)

for x in range(1,len(dataSet)):

temp = []

dataSet[x][0] = int(dataSet[x][0])

trainTargetSet.append(dataSet[x][0])

for y in range(1,785):

#if dataSet[x][y] != 0:

#dataSet[x][y] = 1

dataSet[x][y] = int(dataSet[x][y])

temp.append(dataSet[x][y])

trainDataSet.append(temp)

with open(filename2,'r') as csvfile2:

lines2 = csv.reader(csvfile2)

dataSet2 = list(lines2)

for x in range(1,len(dataSet2)):

temp = []

for y in range(784):

#if dataSet2[x][y] != 0:

#dataSet2[x][y] =