keras有模块可视化工具,可以以图片方式可视化网络结构
from keras.utils import plot_model
使用时只要指定模型和输出的文件即可
plot_model(model,to_file='model.png')
结果(model.png)如下图所示
但是
plot_model(model,to_file='model.png')
语句会报错
keras ImportError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
该错误提示具有误导性
即便执行了
pip install pydot
pip install graphviz
仍会报错
实际上问题在于执行plot_model需要graphviz的二进制文件
解决方法是在graphviz官网下载界面http://www.graphviz.org/download/选择自己操作系统对应的版本的.msi文件
例如Windows的下载链接是https://graphviz.gitlab.io/_pages/Download/windows/graphviz-2.38.msi
安装完成后添加环境变量即可