sudo apt-get install graphviz
https://keras.io/visualization/
系统: ubuntu16.04
前置条件:
依赖 pydot-ng ,若出现错误,用命令行输入
pip install pydot-ng
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.Graph
对象,然后按照自己的需要配置它,例如,如果要在ipython中展示图片
from IPython.display import SVG
from keras.utils.visualize_util import model_to_dot
SVG(model_to_dot(model).create(prog='dot', format='svg'))
【Tips】依赖 pydot-ng 和 graphviz,若出现错误,用命令行输入
pip install pydot-ng & brew install graphviz