我正在尝试使用 plotmodel 绘制 keras 模型。

设置：

我安装了graphviz二进制文件：

choco install graphviz

添加到 bin 文件夹的路径，然后我做了：

pip install pydotplus
pip install graphviz

代码：

当我尝试执行时出现此错误：

from keras.models import Sequential
from keras.layers import Dense
from keras.utils.vis_utils import plot_model
model = Sequential()
model.add(Dense(2, input_dim=1, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
plot_model(model, to_file='model_plot.png', show_shapes=True, show_layer_names=True)

错误：

ImportError: 无法导入 pydot。您必须安装 pydot 和 graphvizpydotprint才能工作。

有人可以建议一些替代方法来绘制 keras 模型吗？我需要为 CNN 实现这个。