Keras Callbacks

在每个training/epoch/batch结束时,如果我们想执行某些任务,例如模型缓存、输出日志、计算当前的auc等等,Keras中的callback就派上用场了。

Example记录每个batch的损失函数值


Keras Callbacks_第1张图片

如上述例子,我们可以继承keras.callbacks.Callback来定义自己的callback,只需重写其中的6个方法即可

on_train_begin

on_train_end

on_epoch_begin

on_epoch_end

on_batch_begin

on_batch_end

可在这6个方法中定义自己想要的属性,通过self.model可以访问模型本身,self.params可以访问训练参数。

可能有用的属性

self.validation_datavalidate数据集

self.validation_data[0]为X

self.validation_data[1]为y


转载自:https://www.jianshu.com/p/eb4e76e239df

你可能感兴趣的:(Keras Callbacks)