Kerasでのモデル可視化

Kerasでのモデル可視化方法は、.summary()とplot_modelを使用する方法の2種類がある。それぞれ使用した結果を比較する。

.summary

まずは、.summary()を試してみる。

print(model.summary())

plot_model

次に、plot_modelを試してみる。

from tensorflow.keras.utils import plot_model
plot_model(model, show_shapes = True, expand_nested = True, to_file='model.png')

結果

plot_modelは図示されるので見やすい。一方で、plot_modelのほうではパラメータの数が見えない。Tensorflowの公式ドキュメントを見ても、表示させるための引数は無さそう。
https://www.tensorflow.org/api_docs/python/tf/keras/utils/plot_model