标签： python tensorflow keras
我已经成功地训练了我的模型，但是现在我需要对很多数据进行分类。要做到这一点，我正在使用predict_generator进行批处理（内存效率高，并且还可以从多处理中加速）。但是我也想在每一批之后保存我的输出，而不是将所有输出收集在一个大数组中。我想这样做，因为它似乎更安全。我敢肯定有一种方法可以对回调进行此操作，但是我找不到使用模型输出而不是诸如损失之类的汇总数据的回调示例
我正在使用Tensorflow 2.0