模型可视化
keras.utils.vis_utils 模块提供了一些绘制 Keras 模型的实用功能(使用 graphviz)。
plot_model
以下实例,将绘制一张模型图,并保存为文件:
from keras.utils import plot_model
plot_model(model, to_file=‘model.png’)
from keras.utils import plot_model
plot_model(model, to_file='model.png')
此外,你也可以直接取得 pydot.Graph 对象并自己渲染它。 例如,ipython notebook 中的可视化实例如下:
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
SVG(model_to_dot(