ニューラルネットワークアーキテクチャを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ファイルはon Githubです。

Answer 1

pydot pythonパッケージをまだインストールしていない場合は、インストールしてください。 pydotの再インストールをお持ちの場合は、問題の解決に役立ちます。

Answer 2

ケラスプロジェクトのissues pageでも問題が参照されています。 find_graphvizがバージョン1.2.0ではremovedだったので、pydot < = 1.1.0のバージョンをインストールする必要があります。代わりに、pydot-ngをインストールすることもできます。これはケラスの開発者がrecommendedです。