自动下载并转化MNIST数据的格式,将数据从原始的数据包中解析成训练和测试神经网络时使用的格式。
自动将MNIST数据集划分为train,validation,test三个数据集。
from tensorflow.examples.tutorials.mnist import input_data
minist = input_data.read_data_sets("/path/to/MNIST_data/", one_hot=True)
print("Training data size:", minist.train.num_examples)
print("Validating data size:", minist.validation.num_examples)
print("Testing data size:", minist.test.num_examples)
print("Example training data label:", minist.train.labels[0])
下载成功的结果为:
image.png
网友评论