python - 如何在 on_epoch_end 中获得 true 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)