plot_model报错的解决方法

from tensorflow.python.keras.utils.vis_utils import plot_model

plot_model(model1, to_file="model1.png", show_shapes=True)

报错

raise ImportError('Failed to import pydot. You must install pydot'
ImportError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.

这里先安装graphviz再安装pydot

之后再运行，仍报错，找到错误位置

pydot.Dot.create(pydot.Dot())

报错信息

FileNotFoundError: [WinError 2] "dot" not found in path.

解决方法

from tensorflow.python.keras.utils.vis_utils import plot_model,model_to_dot
from IPython.display import SVG
SVG(model_to_dot(model1).create(prog='dot', format='svg'))
plot_model(model1, to_file="model1.png", show_shapes=True)

结果