模型可视化

keras.utils.vis_utils模块提供了一些绘制Keras模型的实用功能(使用graphviz)。

以下实例，将绘制一张模型图，并保存为文件：

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

plot_model有两个可选参数:

  • show_shapes (默认为False) 控制是否在图中输出各层的尺寸。
  • show_layer_names (默认为True) 控制是否在图中显示每一层的名字。

此外，你也可以直接取得pydot.Graph对象并自己渲染它。
ipython notebook中的可视化实例如下：

  1. from IPython.display import SVG
  2. from keras.utils.vis_utils import model_to_dot
  4. SVG(model_to_dot(model).create(prog='dot', format='svg'))