如何用keras保存模型的可视化图

2025-12-11 05:27:29

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的下载链接,也可以直接在官网下载。

如何用keras保存模型的可视化图

4、下载之后直接安装就好,可以自己选择安装路径。之后需要将Graphviz2.38的bin目录添加到系统环境变量中。

5、环境变量的位置根据下图的指示。我下载到了E盘,所以path是E:\Graphviz\bin(根据实际情况填写)

如何用keras保存模型的可视化图

如何用keras保存模型的可视化图

如何用keras保存模型的可视化图

如何用keras保存模型的可视化图

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

如何用keras保存模型的可视化图

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