Kerasで学習後に中間層の出力を取り出す方法

TensorFlowでどうやって中間層の出力を取り出すかを調べたけど, Kerasはどうするんだろうって思って調べたら、簡単にできる.

TensorFlowは→TensorFlowで学習後に中間層の出力を取り出す方法

inputのレイヤとoutputのレイヤを指定してModelを作り直すというので終わり.あとはいつも通りpredict関数を呼び出すだけでもう中間層の出力を取り出すことができる.

https://keras.io/getting-started/faq/#how-can-i-obtain-the-output-of-an-intermediate-layer

上記より引用すると以下.

from keras.models import Model

middle_model = Model(inputs=model.input, outputs=model.get_layer(layer_name).output)
#Model(inputs=入力レイヤの入力, outputs=出力レイヤの出力)
middle_output = middle_model.predict(data)
#dataは入力データ

 

 

About the author

コメントを残す