3
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

Kerasのgraphvizモジュールで学習モデルを可視化

Posted at

いろいろ参考にさせてもらったらなんかいろいろ古かったぽいのでメモ。

brew install graphviz
でインストール。

Python3を使ってる場合、pydot_ngをインストールしないとグラフが書けないらしいのでインストール。

pip install pydot_ng

で、プログラム内でKerasのGraphvizを使うためのモジュールを読み込む。グラフの構造を画像で保存したいなら以下。

from keras.utils import plot_model
JupyterNotebookで可視化したいなら以下。

from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot

で、学習モデルを構築後、学習前に
plot_model(model, to_file='model.png')
で出力。元気な学習モデルのグラフが産み落とされた。

3
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
3
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?