运行环境:
win10 + anaconda3-spyder + python3.7.4 + tensorflow2.0.0
首先需要安装两个包pydot和graphviz,不然会报错:
Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
然后去anaconda prompt 里去install,命令如下:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pydot-ng
【注意】:需要注意的是pydot与pydot-ng的区别,pydot是对应python2的旧版本,python-ng是兼容python2与python3的新版本,刚开始就安装的是pydot,可视化的时候依然报错:Failed to import pydot. You must install pydot and graphviz for `pydotprint`。
这样就安装成功了。
接下来,就是要安装了Graphviz。
官网下载一个.msi文件。
(到官网https://graphviz.gitlab.io/download/自己找,win系统可以直接点击下面这个链接:https://graphviz.gitlab.io/_pages/Download/Download_windows.html)
下载完后进行安装。
然后添加到将对应的bin目录的路径添加到系统环境变量中。
学了一招,下面这样也可以添加环境变量?(但是我是老老实实手动添加的)
注意下面这样用的时候,别忘了分号!!
import os
os.environ["PATH"] += ";路径"
检验安装成功:
打开windows命令行界面,输入dot -version
,然后按回车,如果显示graphviz的相关版本信息,则安装配置成功。如图:
两个依赖包都安装成功后,重启spyder,建立model。
接下来,终于可以可视化模型啦。
导入模块,即可使用。(记得重启spyder哦)
from tensorflow.keras.utils import plot_modelplot_model(model, './model11.jpg', show_shapes=True)
ps:看到一篇博客里面写可能会出现这种错误?(如果按照上述过程,依旧未能正常运行的话,就参考一下下面这种情况。)