python kNN算法实现MNIST数据集分类 k值1-120

python kNN算法实现MNIST数据集分类 k值1-120

# -*- coding: utf-8 -*-
“““
Created on Thu Feb 25 14:40:06 2016
load MNIST dataset
@author: liudiwei
“““
import numpy as np 
import struct
import matplotlib.pyplot as plt 
import os

class DataUtils（object）:
	“““MNIST数据集加载
	输出格式为：numpy.array（）    
	
	使用方法如下
	from data_util import DataUtils
	def main（）:
		trainfile_X = ‘../dataset/MNIST/train-images.idx3-ubyte‘
		trainfile_y = ‘../dataset/MNIST/train-labels.idx1-ubyte‘
		testfile_X = ‘../dataset/MNIST/t10k-images.idx3-ubyte‘
		testfile_y = ‘../dataset/MNIST/t10k-labels.idx1-ubyte‘
		
		train_X = DataUtils（filename=trainfile_X）.getImage（）
		train_y = DataUtils（filename=trainfile_y）.getLabel（）
		test_X = DataUtils（testfile_X）.getImage（）
		test_y = DataUtils（testfile_y）.getLabel（）
		
		#以下内容是将图像保存到本地文件中
		#path_tra

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----

     文件       3391  2017-02-28 10:53  kNN\data_util.py

     文件       3500  2017-03-10 11:25  kNN\kNN.py

     目录          0  2017-03-14 10:11  kNN

----------- ---------  ---------- -----  ----

                 6891                    3