如何用keras保存模型的可视化图
1、在进行keras模型保存时需要用到的库有三个,分别使用如下命令行下载:
pip install pydot-ng
pip install graphviz
pip install pydot
2、下载之后,就可以使用代码:
from keras.utils import plot_model
plot_model(model, to_file='model.png', show_shapes=True)
将训练的model结构保存为一张图片
3、但是执行代码的时候会出现如下错误,错误显示没有安装其可执行文件,需要到官网https://www.graphviz.org/下载。在注意事项中给出graphviz-2.38.msi的下载链接,也可以直接在官网下载。

4、下载之后直接安装就好,可以自己选择安装路径。之后需要将Graphviz2.38的bin目录添加到系统环境变量中。
5、环境变量的位置根据下图的指示。我下载到了E盘,所以path是E:\Graphviz\bin(根据实际情况填写)




6、添加完之后重启电脑,再次运行代码就可以看到生成的model图。

声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
阅读量:104
阅读量:82
阅读量:106
阅读量:74
阅读量:148