首页 > 其他 > 详细

tensorflow中导入下载到本地的mnist数据集

时间:2018-11-23 21:20:22      阅读:288      评论:0      收藏:0      [点我收藏+]

mnist数据集获取:可从Yann LeCun教授管网获取;
tensorflow中可使用input_data.read_data_sets("/worker/mnistdata/", one_hot = True) 导入下载到本地的mnist数据集; "/worker/mnistdata/"为数据集存放的位置.

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data  #tensorflow已经包含了mnist案例的数据
mnist = input_data.read_data_sets("/worker/mnistdata/", one_hot = True) #导入已经下载好的数据集,"/worker/mnistdata/"为存放mnist数据集的文件夹

print(mnist.train.images.shape, mnist.train.labels.shape)
print(mnist.test.images.shape, mnist.test.labels.shape)
print(mnist.validation.images.shape, mnist.validation.labels.shape)
Extracting /worker/mnistdata/train-images-idx3-ubyte.gz
Extracting /worker/mnistdata/train-labels-idx1-ubyte.gz
Extracting /worker/mnistdata/t10k-images-idx3-ubyte.gz
Extracting /worker/mnistdata/t10k-labels-idx1-ubyte.gz
(55000, 784) (55000, 10)
(10000, 784) (10000, 10)
(5000, 784) (5000, 10)

tensorflow中导入下载到本地的mnist数据集

原文:https://www.cnblogs.com/forsch/p/10009758.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!