0

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

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

提前致谢!

4

1 回答 1

0

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

于 2021-04-08T19:12:40.310 回答