Keras 回调实例没有属性“set_model”答案

【问题标题】：Keras Callback instance has no attribute 'set_model'Keras 回调实例没有属性“set_model”
【发布时间】：2017-12-13 07:13:36
【问题描述】：

我正在尝试使用 keras 的回调功能记录 Keras 中使用的每个时期的状态。这是回调类的示例代码

class TimingCallback():
    def __init__(self):
        self.logs=[]
    def on_epoch_begin(epoch, logs={}):
        self.starttime=time()
    def on_epoch_end(epoch, logs={}):
        self.logs.append(time()-self.starttime)

这是我的模型。

cb = TimingCallback()
model.fit(X, Y, epochs=150, batch_size=10, callbacks=[cb])

执行时出现以下错误。

错误：

AttributeError: TimingCallback 实例没有属性'set_model'

谁能帮我弄清楚为什么会发生这种情况？

【问题讨论】：

标签： python callback keras

【解决方案1】：

set_model 是keras.callbacks.Callback 中定义的方法。要编写自定义回调，您必须继承 keras.callbacks.Callback。否则，您的回调将缺少一些 Keras 内部使用的必要方法。

将第一行更改为下一行应该可以。

class TimingCallback(keras.callbacks.Callback):

【讨论】：