使用plot_model绘制keras模型图
安装如下依赖:
pip install pydot
pip install graphviz
conda install graphviz python-graphviz
定义完模型后,用如下代码绘制:
from keras.utils import plot_model
plot_model(self.model,to_file="model.png",show_shapes=True)
输出模型图的一部分如下图所示:
定义好网络结构之后,冻结网络结构以防止修改
可使用如下代码:
sess.graph.finalize()