通过模型可视化可以更直观的观察到搭建模型的参数和参数在不同层的传递过程。
1.安装graphviz模块,安装graphviz软件,安装pydot模块,这部分的顺序不能错,在本人的anaconda2环境中是没有自带这些模块的,但是真的需要与否还不确定。但是装上没有任何不良影响。
pip install graphviz
sudo apt-get install graphviz
pip install pydot==1.1.0 # 都说只能装1.1.0版本
2.在主函数中调用模块,首先列出几种测试不通过的版本
from keras.utils.visualize_util import plot
from keras.utils.vis_utils import plot
- 只有下面这种是可以用的,也是官网给的(看官网API得真理)
from keras.utils import plot_model
- 具体用法:
plot_model(temp_model, to_file='model.png',show_shapes='True')
to_file是路径,一般都保存在当前目录下
show_shapes='False'(默认),输出的是简单版的网络结构图
show_shapes='True',输出的是完全版的网络结构图