一、安装所需的graphviz、pydot、pygraphviz
1. 先安装graphviz
pip install graphviz
安装完成后把bin加到环境变量path内
2. 安装pydot或pydot-ng
使用pip 在命令行安装即可。
3. 安装pygraphviz
3.1 在官网下载pygraphviz-1.5.zip
3.2 解压缩到pygraphviz-1.5路径下,然后修改setup.py的57、58两行,如:
include_dirs=['C:\\Program Files (x86)\\Graphviz2.38\\include'],
library_dirs=['C:\\Program Files (x86)\\Graphviz2.38\lib'],
3.3 在pygraphviz-1.5目录下,命令行中运行
pip setup.py install
我的电脑中由于没有visual c++ 14.0,所以无法编译,我选择一个偷懒的方法,直接下载whl文件,下载地址:
https://github.com/CristiFati/Prebuilt-Binaries/tree/master/PyGraphviz/v1.5/Graphviz-2.42.2
下载pygraphviz-1.5-cp37-cp37m-win_amd64.whl后,在命令行使用pip命令直接安装即可
二、使用
安装完成后,如下使用:
from tensorflow.keras.utils import plot_model
plot_model(model, to_file='filename.png')