Ubuntu
Keras
pydot

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

要点

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

これで解決しました。