python - Keras 回调实例没有属性 'set_model'

Question

我正在尝试使用 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 instance has no attribute 'set_model'

谁能帮我弄清楚为什么会这样？

Answer 1

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

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

class TimingCallback(keras.callbacks.Callback):