如何在 on_epoch_end 中获得真正的 y_true 和 y_pred？

Question

我试图在每个时期结束时获得 y_true 和 y_pred。我使用 fit_generator 来训练我的模型。我试图在每个时代之后找到特异性。

class LossHistory(keras.callbacks.Callback):
    def on_epoch_end(self, epoch, logs={}):
      y_true = ?
      y_pred = ?

loss_history = LossHistory

fit_generator(train_data, .. callbacks=[loss_history])

Answer 1

默认情况下您无权访问此信息，但您可以通过将它们传递给构造函数来提供Callback 任何您想要的属性。例如，如果您使用的是生成器（数据存储在 generator.y 属性中）：

class LossHistory(keras.callbacks.Callback):

    def __init__(self, data_generator, **kwargs):
        self.generator = data_generator
        Super(LossHistory, self).__init__(**kwargs)

    def on_epoch_end(self, epoch, logs={}):
        y_true = self.generator.y
        y_pred = self.model.predict_generator(self.generator)


loss_history = LossHistory(train_data_generator)