Keras 对 tf.data 数据集的预测不会停止

Question

我在使用生成器对我的模型进行预测时遇到了问题。 根据documentation，如果steps = None 预测应该贯穿整个数据集。 但相反，它通过了它，我重新开始。我在图像数据集上使用flow_from_dataframe。

我应该将步长固定为ceil(len(dataset) / batch_size)吗？

提前致谢！

Answer 1

嗯，这不是一个真正的解决方案，但是将步长固定到数据集的长度就可以了。 我想文档必须更新。