tensorflow(6) mnist.train.next_batch()函数解析

最新推荐文章于 2025-03-25 20:14:31 发布

lgy_keira

最新推荐文章于 2025-03-25 20:14:31 发布

阅读量5w

点赞数 16

分类专栏： tensorflow

本文链接：https://blog.youkuaiyun.com/u013608336/article/details/78747102

版权

tensorflow的feed_dict原理，它需要的每次生成一个batch的数据。

1. Dataset类

将数据处理部分写成一个类，init函数中定义了一些参数

class DataSet(object):

  def __init__(self,
               images,
               labels,.....)
    self._images = images
    self._labels = labels
    self._epochs_completed = 0 # 已经经过了多少个epoch
    self._index_in_epoch = 0 # 在一个epoch中的index