資源簡介
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
評論
共有 條評論