python - 使用 keras 1.2 的神经网络模型可视化

Question

关注此keras doc ，我执行以下操作以在 Mac 中可视化模型

from keras.utils import plot_model
plot_model('model.h5', to_file='model.png')

但是出现错误

Traceback (most recent call last):
  File "model.py", line 219, in <module>
    plot_model('model.h5', to_file='model.png')
NameError: name 'plot_model' is not defined

我查了一下互联网，发现了一个 possible solution 。但这并不完全有效。

例如，我将 graphviz 安装在 /Users/MYNAME/miniconda3/envs/carnd-term1/lib/graphviz而不是<path to anaconda environment>\Library\bin\graphviz\正如可能的解决方案中提到的。所以我不确定我是否在 PATH 变量中添加了正确的路径

我还需要修改 <path to anaconda environment>\lib\site-packages\keras\utils\visualize_util.py文件但没有site-packages lib 内的目录我的目录miniconda3环境。

请提出建议。

Answer 1

您使用的是过时版本的 Keras。对于 Keras 1.2，函数调用为 keras.utils.visualize_util.plot()。此外，您还需要为该函数提供一个模型对象，而不是它的文件名。使用 model = load_model('model.h5') 并将此对象提供给 visualize_util.plot()。