如何编写回调来从Keras的fit_generator（）获取预测？

Question

毫无疑问，我们可以定义自己的回调来从keras中的fit_generator中丢失批量，如下所示：

class LossHistory(Callback):
    def on_train_begin(self, logs={}):
        self.losses = []
    def on_batch_end(self, batch, logs={}):
        self.losses.append(logs.get('loss'))

我想知道是否有一种方便的方法来批量处理每个样本的预测，就像每批次丢失一样。

我已经搜索了很长时间的解决方案，但没有任何提示，非常感谢你的帮助！