Keras输出模型图片

官网指导手册的输出方法好像不能用了（下图是指导手册的方法），我的现在Keras版本是2.8，Python3.9，按照这个运行会报错。

keras.utils.plot_model(model, "my_first_model.png")

解决：

from keras.utils.vis_utils import plot_model

plot_model(model,to_file='model.png',show_shapes=True)

要变成如上的import语句，就可以正常使用了。

另外要安装pydot（pip install pydot）和 graphviz： https://graphviz.gitlab.io/download/，下载选择exe安装包可选择自动添加PATH路径（D:\Graphviz\bin），搞定后cmd下测试dot -c，没报错就重启python运行代码。