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

【问题标题】：How can a neural network architecture be visualized with Keras?如何使用 Keras 可视化神经网络架构？
【发布时间】：2016-07-18 16:50:09
【问题描述】：

我尝试了以下方法：

#!/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 文件可以在on Github找到。

【问题讨论】：

另见：How do you visualize neural network architectures?

标签： python neural-network visualization keras

【解决方案1】：

keras项目的issues page上也提到了这个问题。您需要安装pydot find_graphviz 在版本1.2.0 中是removed。或者，您可以安装 pydot-ng，它是由 keras 开发人员提供的 recommended。

【讨论】：

【解决方案2】：

如果您还没有安装pydot python 包 - 尝试安装它。如果您有 pydot 重新安装应该可以帮助您解决问题。

【讨论】：

pydot 安装在版本 1.2.2 中。重新安装它没有帮助。