如何使用Keras可视化神经网络架构？

Question

我尝试了以下内容：

#!/usr/bin/env python

import keras
from keras.models import model_from_yaml

model_file_path = 'model-301.yaml'
weights_file_path = 'model-301.hdf5'

# Load network
with open(model_file_path) as f:
    yaml_string = f.read()
model = model_from_yaml(yaml_string)
model.load_weights(weights_file_path)
model.compile(optimizer='adagrad', loss='binary_crossentropy')

# Visualize
from keras.utils.visualize_util import plot

然而，这会出现错误：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/moose/.local/lib/python2.7/site-packages/keras/utils/visualize_util.py", line 7, in <module>
    if not pydot.find_graphviz():
AttributeError: 'module' object has no attribute 'find_graphviz'

我该如何修复这个问题？

注意：hdf5和YAML文件可以在Github上找到。

- Martin Thoma

---

参见：你如何可视化神经网络架构？ - Martin Thoma

Answer 1

问题还在Keras项目的问题页面上有所提及。 您需要安装版本为pydot <= 1.1.0，因为函数find_graphviz在版本1.2.0中被删除了。或者，您可以安装推荐的pydot-ng代替。这是由Keras开发人员建议的。

- sietschie

Answer 2

如果您尚未安装pydot python包，请尝试安装它。如果您已经安装了pydot，重新安装可能有助于解决问题。

- Marcin Możejko

---

pydot 已安装版本为 1.2.2。重新安装并没有帮助。 - Martin Thoma