美文网首页TensorFlow技术帖
Python加载tensorflow mnist数据集

Python加载tensorflow mnist数据集

作者: 李小西033 | 来源:发表于2017-11-30 11:06 被阅读149次

使用pip安装tensorflow

python3 -m pip install tensorflow

使用tensorflow提供的input_data.read_data_sets方法载入mnist数据集,并将其保存在项目的data目录中(该目录依据情况指定即可)

# coding:utf-8
from tensorflow.examples.tutorials.mnist import input_data

# input_data.read_data_sets 将mnist自动划分为train,validation和test数据集
mnist = input_data.read_data_sets("../data",one_hot=True)

print("Training data size:",mnist.train.num_examples)

print("Validation data size:",mnist.validation.num_examples)

print("Testing data size:", mnist.test.num_examples)

print("Example training data:",mnist.train.images[0])

print("Example training data label:",mnist.train.labels[0])

相关文章

网友评论

    本文标题:Python加载tensorflow mnist数据集

    本文链接:https://www.haomeiwen.com/subject/uoxvbxtx.html