使用plot_model得先安装好另外两个库,graphviz和pydot
pip install graphviz
pip install pydot
再安装软件graphviz.smi,下载地址,https://graphviz.gitlab.io/_pages/Download/Download_windows.html
之后再导入库
from keras.utils.vis_utils import plot_model
或
from keras.utils import plot_model
如果这两个都不行的话,并出现以下错误:
TypeError: ‘InputLayer’ object is not iterable
那尝试这个导入语句:from tensorflow.keras.utils import plot_model
可以成功了!