keras画神经网络拓扑结构图

keras画神经网络拓扑结构图

步骤1： 安装graphviz
https://graphviz.gitlab.io/download/
进入该网站

下载后双击

下一步

gogogo

这里路径可以修改，但是要记好，尽量避免中文路径

安装安装安装

步骤2：这里前边步骤里已经选了添加环境变量，但最好还是看看系统中是否有环境变量，以下步骤是针对没有环境变量的情况。

此电脑–右击属性—高级系统设置

按步骤将安装所在路径添加至系统环境变量

进入pycharm,在终端输入：
pip install pydot-ng
pip install graphviz
pip install pydot

安装后去setting检查下。

在interpreter下检查上述三者是否安装
如果没有安装，点击加号搜索后安装

步骤3：进入代码页，在开头添加如下代码，路劲换成自己的
呦。

from keras.utils import plot_model
import os
os.environ["PATH"] += os.pathsep + 'D:/graphviz/bin/'

在模型构建完成后的代码处添加如下代码，我一般是放在fit函数之后。

#添加如下代码
plot_model(encoder, to_file='./model.png', show_shapes=True)

运行即可生成model.png

bingo!!!