美文网首页
Keras Callbacks

Keras Callbacks

作者: 丫头_631e | 来源:发表于2019-11-06 10:08 被阅读0次

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

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

    如上述例子,我们可以继承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

          本文链接:https://www.haomeiwen.com/subject/yznybctx.html