Edited at

Ubuntu環境でkeras使ってplot_modelしようとしたらエラーになったので対処したお話

More than 1 year has passed since last update.


要点

ubuntu14.04上でkerasのplot_modelを成功させるまでの手順をまとめます

(使用しているPythonのバージョンは3.6.3)


状況

kerasでモデルの可視化ができるということが分かったので、さっそく自分の学習モデルを可視化すべく以下のコードを追加。

from keras.utils import plot_model

plot_model(model, to_file='model.png', show_shapes=True)
# show_shapesオプション→Trueで各層のデータのshape情報も追加される


しかし

エラーが出た。

pydotを入れろ、というエラーだったので

pip install pydot

でインストール。


しかs(ry

またもエラーが出た。

どうやらpydotだけでは不十分なよう。


結局


  1. graphvizというパッケージが必要

  2. pydotはpythonのバージョンによっては使えない(現在は開発されていないようです 3.5系、3.6系はNGのようです)

1に関してはGraphvizというパッケージをインストール。

pipコマンドを使ってpython用のラッパーと、apt-get でパッケージ本体をインストール。

pip install graphviz

sudo apt-get install graphviz

2に関してはpydotの代替としてpydotplusをインストール。

pip install pydotplus

これで解決しました。