基于keras 1.2的神经网络模型可视化

Question

遵循这个角化病，下面我将在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

我查看了互联网，发现了一个可能解决办法。但这不太管用。

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

另外，我需要修改<path to anaconda environment>\lib\site-packages\keras\utils\visualize_util.py文件，但是在site-packages环境的lib目录中没有site-packages目录。

请建议一下。

Answer 1

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