我们在跑模型的时候,都想输出一下模型的大致形状,keras提供了一个很好的接口,通过短短的几行代码就行了,首先需要安装必
要的包:
pip install pydot
pip install graphviz
在model,定义好以后,输入下面的代码可以输出了:
from keras.utils import plot_model
plot_model(model,show_shapes=True,to_file="model.png")
参考文献
[1].keras模型可视化plot_model模块调用报错. https://blog.csdn.net/da_kao_la/article/details/80025124
[2].模型可视化. https://keras-cn.readthedocs.io/en/latest/other/visualization/