我在使用生成器为我的模型进行预测时遇到了问题。根据文档,如果steps = None
predict 应该贯穿整个数据集。但相反,它通过了它,我重新开始。我正在使用flow_from_dataframe
图像数据集。
我应该将步长固定为ceil(len(dataset) / batch_size)
吗?
提前致谢!
我在使用生成器为我的模型进行预测时遇到了问题。根据文档,如果steps = None
predict 应该贯穿整个数据集。但相反,它通过了它,我重新开始。我正在使用flow_from_dataframe
图像数据集。
我应该将步长固定为ceil(len(dataset) / batch_size)
吗?
提前致谢!