Keras中模型架构可视化 亲测有效

文章目录
Keras中模型架构可视化

1. 安装相关Python package

2. 安装graphviz

3. 设置graphviz环境变量

4. 参考样例

5. 安装相关Python package
   pip install graphviz
   pip install pydot
   pip install pydot-ng
   1
   2
   3

6. 安装graphviz
     graphviz官网

7. 设置graphviz环境变量
   注：graphviz使用默认安装位置

选择“此电脑”，右键“属性”，进入“系统”界面；选择“高级系统设置”，进入“系统属性”界面；单击右下角“环境变量(N)”，进入“环境变量”界面。

创建用户环境变量

变量名：GRAPHVIZ_INSTALL_DIR

变量值：C:\Program Files (x86)\Graphviz2.34

创建系统环境变狼

变量名：GRAPHVIZ_DOT

变量值：C:\Program Files (x86)\Graphviz2.34\bin\dot.exe

加入系统变量 - Path

双击系统变量Path, 添加路径C:\Program Files (x86)\Graphviz2.34\bin

4. 参考样例
   from keras.utils import plot_model
   from keras import models
   from keras import laters

   Create model

   model = models.Sequential()
   model.add(layers.Dense(32, activation='relu))
   model.add(layers.Dense(1))
   model.compile(optimizer=‘adam’, loss=‘mse’)

   Plot model

   plot_model(model,to_file=‘model1.png’, show_shapes=True)
   1
   2
   3
   4
   5
   6
   7
   8
   9
   10
   11
   12
   注：若出现pydot can not find…

修改pydot.py。在pydot.py中找到类Dot，修改self.prog = 'dot’为self.prog = ‘dot.exe’。
然后再模型重启即可