在keras可视化模型输出时,往往用:
from keras.utils import plot_model
plot_model(model, to_file=‘model.png’)
但是这个函数需要配置一系列的环境。
首先,确保环境中有pydot和graphviz。安装方法:
pip install pydot
pip install graphviz
如果是Windows的话,需要安装graphviz的客户端,安装方法见博客:《Graphviz 安装》。注意在安装时,务必要将graphviz的目录添加到环境变量中,一般来讲,到这里就没问题了。但是如果还是报错的话,也就是报错信息:
OSError: `pydot` failed to call GraphViz.Please install GraphViz (https://www.graphviz.org/) and ensure that its executables are
解决方法有两个:
- 尝试重启,可能是添加的目录还没有被识别
- 在运行的代码前面人为添加graphviz的路径,也就是添加如下代码:
import os
os.environ["PATH"] += os.pathsep + 'D:/Software/Graphviz/bin'
问题就成功解决了!