Keras plot_model问题解决办法

from keras.utils.vis_utils import plot_model

直接调用keras中的可视化函数会报错,原因是keras源代码中出现了问题,我用的操作系统是window 10,下面的操作可以解决plot_model的问题。。
安装附加依赖项
pydot
pip install pydot_ng
官方文档中说直接安装pydot,但是由于keras中使用了pydot.find_graphviz()函数,而这一函数在1.2.*后的版本被剥离了,所以不能直接pip安装pydot,而是安装pydot_ng.
在keras的visualize_util.py中,import pydot或pydot_ng的源码如下:

这里写图片描述
graphviz
• pip install graphviz
• 到http://www.graphviz.org/Download_windows.php 中下载对应的msi并安装
• 修改 D:*\python3.5\Lib\site-packages\pydot_ng__init__.py中的代码,使pydot找到graphviz
o 注释掉 这里写图片描述下面的if中的部分
o Method 2 不做修改
o Method3 添加
path = r”D:\software\graphviz\bin”
该路径指向刚才下载的msi的安装路径中的bin

源代码修改好后,可以检验下:

from keras.utils.vis_utils import plot_model
plot_model(model,to_file = filepath + filename,show_shapes = True)

猜你喜欢

转载自blog.csdn.net/baidu_36161077/article/details/81058404