fit()训练日志信息history源码分析

fit_generator()的返回值是这个model.history,事实上他就是一个普通的回调函数。在fit_generator()方法中,自动加入了一个回调函数History(),并且这个Histor()是默认参数的。

 # prepare callbacks
    model.history = cbks.History()
    _callbacks = [cbks.BaseLogger(
        stateful_metrics=model.stateful_metric_names)]
    if verbose:
        _callbacks.append(
            cbks.ProgbarLogger(
                count_mode='steps',
                stateful_metrics=model.stateful_metric_names))
    _callbacks += (callbacks or []) + [model.history]
    callbacks = cbks.CallbackList(_callbacks)

到callbacks.py可以找到History()的源码:

class History(Callback):
    """Callback that records events into a `History` object.

    This callback is automatically applied to
    every Keras model. The `History` object
    gets returned by the `fit` method of models.
    """

    def on_train_begin(self, logs=None):
        self.epoch = []
        self.history = {}

    def on_epoch_end(self, epoch, logs=None):
        logs = logs or {}
        self.epoch.append(epoch)
        for k, v in logs.items():
            self.history.setdefault(k, []).append(v)

说明history只在每个epoch结束时记录了模型信息,并没有在每个iteration记录。我们要想获取更详细的训练信息,可以直接自己继承Callback写回调函数来获取想要的数据。

去看了一下Keras中文文档,其实官方也已经给出这样的说明了
fit()训练日志信息history源码分析_第1张图片

你可能感兴趣的:(学习笔记)