如果直接使用
from keras.utils import plot_model报错可以试试
from tensorflow.python.keras.utils import plot_model
如果还是报错说你未安装以下两个包,
Failed to import pydot. You must install pydot and graphviz for ‘pydotprint‘ to work
那么就安装这两个包
首先是pydot,这是一个比较过时的包适用于Python3以下版本,如果你是Python3版本要安装python_ng,这个包兼容Python2和3,
pip install python_ng
然后是graphviz这个包不能直接pip要去这个网址下载win版本的https://graphviz.gitlab.io/download/
然后找到Development Windows install packages点击进去,再依次点击
10/ -> cmake/ -> Release -> x64 -> graphviz-install-2.44.2~dev.20201009.0508-win64.exe
下载完以后解压安装,安装时候注意选择为所有用户添加到环境变量
然后如果还是报错,是因为你的graphviz安装时候在c盘需要管理员权限,那么就关闭pycharm以管理员权限重新打开,然后再运行应该就可以了。
顺便一说运用plot_model时模型定义必须指定输入维数input_shape
否则就会报错You must provide an `input_shape` argument.