keras modelの可視化
kerasで実装したmodelを可視化する
準備
graphvizをインストール
$ sudo apt-get install graphviz
その後pydotをインストール
$ pip install pydot graphviz
実装
2行ソースを追加
今回は以下で公開されるmnist_cnnのサンプルコードを使用
GitHub - fchollet/keras: Deep Learning library for Python. Runs on TensorFlow, Theano, or CNTK.
1 import文を追加
from keras.utils import plot_model
2 model.compileの後ろに、plot_modelを追加
plot_model(model, to_file='mnist_cnn.png',show_shapes=True)
結果
できた