PyCharm Tensorflow Plot_model 无法工作

PyCharm Tensorflow Plot_model cannot work

我在 PyCharm.

中使用 TensorFlow 显示绘图模型时遇到问题

我从 设置 部分的 Python 解释器 安装了 pydotgraphviz =28=]PyCharm.

接下来,我 运行 代码并得到一个定义如下的错误。

我该如何解决?

这是我的代码片段,如下所示。

from tensorflow.keras.utils import plot_model

# Plot Model
plot_model(model, show_shapes=True)

这是我的错误,如下所示。

ImportError: ('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.

这是我的解决方案:

1 ) 从 https://graphviz.org/download/

下载 graphviz

2 ) 定义 path 以从 Tensorflow

中识别 plot_model
import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files/Graphviz/bin/'

3 ) 如下所示定义 plot_model。

# Plot Model
plot_model(model, "model.png" , show_shapes=True, show_layer_names=True)