使用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
可以成功了!
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:keras plot_model模块安装 - Python技术站