文章目录
1. keras模型可视化函数
keras.utils.vis_utils模块提供了可视化Keras模型的函数plot_model,可将模型summary信息以图片形式输出。
使用方式如下:
from keras.utils import plot_model
plot_model(model, to_file='model.png')
plot_model 有 4 个可选参数:
- show_shapes (默认为 False) 控制是否在图中输出各层的尺寸。
- show_layer_names (默认为 True) 控制是否在图中显示每一层的名字。
- expand_dim(默认为 False)控制是否将嵌套模型扩展为图形中的聚类。
- dpi(默认为 96)控制图像 dpi
2. 问题解决
实际使用过程中发现,plot_model函数依赖 pydot-ng 和 graphviz包,若直接运行会报错