一、基本环境
Windows7、Anaconda3
二、安装pydot和GraphViz
1.安装GraphViz
在cmd中输入conda install graphviz,安装GraphViz。
2.安装GraphViz软件本身
进入https://graphviz.gitlab.io/_pages/Download/Download_windows.html下载graphviz-2.38.msi,解压后安装。
进入GraphViz安装路径,找到bin文件路径,我的是D:/Graphviz/bin/。将该路径添加到环境变量(我是用户变量和系统变量都添加了)。
如果后面报了错误,可能是手动添加环境变量不起作用,就在Spyder中运行下列程序:
import os
os.environ["PATH"] += os.pathsep + 'D:/Graphviz/bin/'
3.安装pydot
在cmd中运行conda install pydot,安装pydot。
4.安装顺序不能变,按步骤安装即可。
三、使用plot_model
from keras.utils import plot_model
# show_shapes:指定是否显示输出数据的形状,默认为False;show_layer_names:指定是否显示层名称,默认为True
modelstruct_path='C:/Users/Administrator/.keras/model.png' # 路径
plot_model(model, to_file=modelstruct_path,show_shapes=True,show_layer_names=True) # 保存模型