Keras 可视化 model visualization

https://keras.io/visualization/

系统： ubuntu16.04

前置条件：

sudo apt-get install graphviz

依赖 pydot-ng ，若出现错误，用命令行输入

pip install pydot-ng

keras.utils.vis_utils模块提供了画出Keras模型的函数（利用graphviz）

该函数将画出模型结构图，并保存成图片：

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

plot_model接收两个可选参数：

• show_shapes：指定是否显示输出数据的形状，默认为False
• show_layer_names:指定是否显示层名称,默认为True

我们也可以直接获取一个pydot.Graph对象，然后按照自己的需要配置它，例如，如果要在ipython中展示图片

from IPython.display import SVGfrom keras.utils.visualize_util import model_to_dotSVG(model_to_dot(model).create(prog='dot', format='svg'))

【Tips】依赖 pydot-ng 和 graphviz，若出现错误，用命令行输入 

pip install pydot-ng & brew install graphviz