2017-09-19 57 views
1

在此之后keras doc,我下面形象化在我的Mac可视化使用keras 1.2

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文件,但存在lib目录我miniconda3环境内没有site-packages目录。

请建议。

回答

3

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