keras模型可視化及解決'Failed to import pydot'問題

１．keras模型可視化

keras.utils.vis_utils模塊提供了畫出Keras模型的函數（利用graphviz）

該函數將畫出模型結構圖，並保存成圖片：

from keras.utils import plot_model
plot_model(model, to_file='model.png')

plot_model接收兩個可選參數：

• show_shapes：指定是否顯示輸出數據的形狀，默認為False
• show_layer_names:指定是否顯示層名稱,默認為True

 

２．需要安裝pydot-ng 和 graphviz依賴庫

分別sudo pip install之后還是會報錯 Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work，可以通過 sudo apt-get install graphviz 解決．

其它方法可參考：https://stackoverflow.com/questions/36886711/keras-runtimeerror-failed-to-import-pydot-after-installing-graphviz-and-pyd