keras: plot_model安装以及使用

报错cannot import name 'plot_model' from 'keras.utils' (D:\Anaconda\envs\tf2\lib\site-packages\keras\utils__init__.py)

from keras.utils import plot_model
plot_model(model, "model.png");

想知道plot_model的其他功能可以使用

help (plot_model)

但是安装好keras后不能直接使用plot_model。还需要做以下几件事：

1.安装库

pip3 install pydot
#或者是pip3 install pydotplus
pip3 install graphviz
pip3 install pydot_ng

2.下载软件graphviz-2.38.msi

3.添加路径

安装以下两个链接的方法添加路径 ，任选一个即可。

import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'

4.若还是无法使用

4.3可能是安装版本的原因（我安装的时候就是这种原因）
将from keras.utils import plot_model改为from keras.utils.vis_utils import plot_model即可。

我用的是tensorflow2版本以上，python3.9。

Original: https://blog.csdn.net/pxyp123/article/details/123210659
Author: pxyp123
Title: keras: plot_model安装以及使用