Kerasコールバックインスタンスに「set_model」属性がありません

Question

ケラスで使用される各エポックのステータスをケラスのコールバック機能を使って記録しようとしています。これはコールバッククラスのサンプルコードです

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):