如何批量获取mnist数据集

通过mnist里面的一个方法:next_batch方法获取;

代码如下:

from tensorflow.examples.tutorials.mnist import input_data

mnist = input_data.read_data_set('data\',one_hot=True)


example_batch,label_batch = mnist.next_batch(batch_size)

但是这种方法需要用迭代的方式来获取

for i in range(total_batch_num):

    example_batch,label_batch = mnist.next_batch(batch_size)

这样就可以了


你可能感兴趣的:(python,tensorflow)