文章目录
Keras中模型架构可视化
-
安装相关Python package
-
安装graphviz
-
设置graphviz环境变量
-
参考样例
-
安装相关Python package
pip install graphviz
pip install pydot
pip install pydot-ng
1
2
3 -
安装graphviz
graphviz官网 -
设置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
-
参考样例
from keras.utils import plot_model
from keras import models
from keras import latersCreate 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’。
然后再模型重启即可