Kerasモデルのすべてのエポック後にウェイトをチェックする方法

Question

私はKerasでシーケンシャルモデルを使用しています。私は、各エポック後にモデルの重量をチェックしたいと思います。それをどうやって行うのか教えてください。

model = Sequential() 
model.add(Embedding(max_features, 128, dropout=0.2)) 
model.add(LSTM(128, dropout_W=0.2, dropout_U=0.2)) 
model.add(Dense(1)) 
model.add(Activation('sigmoid')) 
model.compile(loss='binary_crossentropy',optimizer='adam',metrics['accuracy']) 
model.fit(X_train, y_train, batch_size=batch_size, nb_epoch=5 validation_data=(X_test, y_test)) 

ありがとうございます。

Answer 1

あなたが探しているものはCallBackです。コールバックは、キーポイントでのトレーニング中に繰り返し呼び出されるKeras関数です。バッチ、エポック、またはトレーニング全体の後にすることができます。 docについてはhere、既存のコールバックのリストを参照してください。

必要なものは、LambdaCallBackオブジェクトを使用して作成できるカスタムコールバックです。

from keras.callbacks import LambdaCallback 

model = Sequential() 
model.add(Embedding(max_features, 128, dropout=0.2)) 
model.add(LSTM(128, dropout_W=0.2, dropout_U=0.2)) 
model.add(Dense(1)) 
model.add(Activation('sigmoid')) 

print_weights = LambdaCallback(on_epoch_end=lambda batch, logs: print(model.layers[0].get_weights())) 

model.compile(loss='binary_crossentropy',optimizer='adam',metrics['accuracy']) 
model.fit(X_train, 
      y_train, 
      batch_size=batch_size, 
      nb_epoch=5 validation_data=(X_test, y_test), 
      callbacks = [print_weights]) 

上記のコードは、すべてのエポックの終わりに、あなた埋め込む重みmodel.layers[0].get_weights()を印刷する必要があります。これは