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中文文档,其实官方也已经给出这样的说明了