我使用tf.train.shuffle_batch()
和tf.train.batch()
来读取.record文件,每个类有两个类和三个示例。如果我想要两批三个例子,我将num_epochs = 1
放在string_input_producer
中,这样我就可以得到两批三个例子,但每次调用它们时,两个函数都会返回两个洗牌批次。 / p>
这两个功能有什么区别?
谢谢!
答案 0 :(得分:1)
tf._train.shuffle_batch()
会以随机顺序返回记录,而tf.train.batch()
会按顺序从您的来源返回记录。根据您要完成的任务,一个或另一个可能更有帮助。我会说shuffle_batch
,因为它为学习过程增加了额外的随机性,但首先要考虑你的任务是否有意义。