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

时间:2017-09-19 02:22:07

标签: python keras

keras doc之后,我会在下面看到我的Mac中的模型可视化

@include media-breakpoint-up(xs) { ... }
@include media-breakpoint-up(sm) { ... }
@include media-breakpoint-up(md) { ... }
@include media-breakpoint-up(lg) { ... }
@include media-breakpoint-up(xl) { ... }

// Example usage:
@include media-breakpoint-up(sm) {
  .some-class {
    display: block;
  }
}

但是得到错误

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

我查看了互联网并找到了possible solution。但这不太有效。

例如,我在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变量中添加了正确的路径

此外,我需要修改<path to anaconda environment>\Library\bin\graphviz\文件,但<path to anaconda environment>\lib\site-packages\keras\utils\visualize_util.py环境的site-packages目录中没有lib目录。

请建议。

1 个答案:

答案 0 :(得分:3)

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