keras的内置函数keras.utils.vis_utils.plot_model可以直接将搭建的神经网络用流程图表示出来
该函数原来为keras.utils.visualize_util.plot,在2017年3月1日的更新中作了修改
visualize_util改为vis_utils;plot改为plot_model;
使用原函数名会报错import error
使用该函数需要额外的函数库和软件,以下说明引自博客:
https://www.jianshu.com/p/3a97c0a43beb
1. (使用pip install)安装pydot和graphviz两个函数库
2. 安装graphviz软件:
https://graphviz.gitlab.io/_pages/Download/Download_windows.html
3. 将路径加入系统变量,从我的电脑-属性-高级系统设置-环境变量 进入
(图片为引用,根据安装路径调整)
4. 检查graphviz是否安装成功,进入windows命令行,输入命令:dot -version,若成功显示版本信息,则认为安装和添加路径成功
5. 引用函数需要以下语句
from keras.utils.vis_utils import plot_model
import os
os.environ["PATH"] += os.pathsep +‘D:/Program Files (x86)/Graphviz2.38/bin/‘ # 后一段路径改为自己安装的路径
6. 绘图语句如下:
plot_model(model,to_file=‘name.png‘,show_shapes=True)
结果如图所示:
Keras.utils.vis_utils.plot_model 绘制神经网络结构示意图
原文:https://www.cnblogs.com/travelcat/p/11429437.html