如何编写回调以从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'))

我想知道是否有一种简便的方法可以像批量获取损失的方式那样批量获取每个样本的预测 。

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