笔者版本为:Python 3.7,Pycharm IDE
在使用前需要安装:
Graphviz软件: https://graphviz.gitlab.io/_pages/Download/Download_windows.html下载对应的msi
编译器的 grapphviz
编译器的 pydot
下载Graphviz后,复制下载路径直到bin,如 ‘C:/Program Files (x86)/Graphviz2.38/bin/’
添加至环境变量 Path 中
代码:
import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'#你的下载路径
from keras.utils.vis_utils import plot_model
plot_model(model, show_shapes=True, to_file='model.png')
备注
运行后可能会报错,选择报错中的pydot.py文件,将其中的第1712行中
self.prog = 'dot'
改为
self.prog = 'dot.exe'
据说对问题解决有帮助
如果不是为了画图,使用summary即可
model.summary()