解决Win10,Keras中plot_model的问题

笔者版本为: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()

你可能感兴趣的:(Keras)