Keras中提供了一个神经网络可视化的函数plot,并可以将可视化结果保存在本地。plot使用方法如下:
from keras.utils.visualize_util import plot
plot(model, to_file='model.png')
注:笔者使用的Keras版本是1.0.6
不过这项功能依赖于graphviz模块与pydot模块,因此需要先安装这两个模块,并安装graphviz软件本身(笔者安装的版本为2.38)。
安装步骤
- 命令行输入
pip install graphviz
- 安装graphviz软件。官网地址为http://www.graphviz.org/
- 解压版:配置环境变量。将安装目录中的
graphviz-2.38\relea
- 解压版:配置环境变量。将安装目录中的