解决(‘You must install pydot (`pip install pydot`) and install graphviz (see…) ‘, ‘for plot_model..
目录
一、报错提示
调用keras.utils.plot_model报错提示:
('You must install pydot (pip install pydot
) and install graphviz (see instructions at https://graphviz.gitlab.io/download/) ', ‘for plot_model/model_to_dot to work.’)
因为tensoerflow2.0以上兼容了keras,只是产生图而已,所以不必在这里使用tf.keras,再者使用tf.keras也会报错,tf的库里没有这个函数。
二、解决方案(共四步)
2.1 安装pydot
按照提示,安装pydot
pip install pydot
2.2 下载并安装graphviz包
踩坑:千万别pip install graphviz。因为测试过无效。
正确做法是从官网
找到如下图所示的Win10 64位版本的graphviz进行下载,然后安装
下一步就好了
2.3 配置环境变量(graphviz)
进入到安装的bin目录下
变量名:别与系统的专用词冲突即可
变量值:路径,我这里是D:\Graphviz\bin (仅供参考)
2.4 调用
添加环境变量后,引入包
from keras.utils.vis_utils import plot_model
模型编译后调用plot_model函数(见下图)
(to_file中设置产生图片的路径,下面这种方式表示保存到当前文件下)
keras.utils.plot_model(model, to_file='graph.png', show_shapes=True)
三、结果展示
模型编译后运行就会产生对应的可视化结构
由于比较长,只截取部分
来源:sinysama