在Keras的fit_generator中是否使用了所有列车样品?

时间:2017-10-12 04:03:41

标签: python tensorflow machine-learning keras neural-network

我正在使用model.fit_generator()来训练Keras的神经网络。在拟合过程中,我将steps_per_epoch设置为16(len(training samples)/batch_size)。

如果迷你批量大小设置为12,并且训练样本总数为195,是否意味着在训练阶段不会使用3个样本?

1 个答案:

答案 0 :(得分:2)

不,因为它是一个生成器,模型不知道训练样本的总数。因此,它在到达使用steps_per_epoch参数定义的最后一步时完成一个纪元。在你的情况下,它确实会在每个时期训练192个样本。

如果您想使用模型中的所有样本,您可以使用参数shuffle在每个纪元的开头随机播放数据。