根据我上一篇文章下载完MNIST数据集后,下一步就是看看keras是如何对它进行分类的。
参考博客:
http://blog.csdn.net/vs412237401/article/details/51983440
之际复制该blog中的代码发现在我这儿运行不通,初步判断是因为Window和Linux系统路径方面差别,处理有点儿问题,所以对此修改了一点
先看原文:
- def load_mnist(path, kind='train'):
- """Load MNIST data from `path`"""
- labels_path = os.path.join(path, '%s-labels-idx1-ubyte' % kind)
- images_path = os.path.join(path, '%s-images-idx3-ubyte' % kind)
- with open(labels_path, 'rb') as lbpath:
- magic, n = struct.unpack('>II', lbpath.read(8))
- labels = np.fromfile(lbpath, dtype=np.uint8)
- with open(images_path, 'rb') as imgpath:
- magic, num, rows, cols = struct.unpack(">IIII", imgpath.read(16))