Wie kann eine neuronale Netzwerkarchitektur mit Keras visualisiert werden?

Question

Ich habe versucht, die folgenden:

#!/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 

jedoch ergibt dies einen Fehler:

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' 

Wie kann ich dieses Problem beheben?

Hinweis: Die Datei hdf5 und die YAML-Datei finden Sie unter on Github.

Answer 1

Wenn Sie das Python-Paket pydot noch nicht installiert haben, versuchen Sie es zu installieren. Wenn Sie pydot Neuinstallation haben, sollte mit Ihrem Problem helfen.

Answer 2

Das Problem wird auch auf die issues page des Keras-Projekts verwiesen. Sie müssen eine Version von pydot < = 1.1.0 installieren, da die Funktion find_graphviz in der Version 1.2.0 removed war. Alternativ können Sie stattdessen pydot-ng installieren, was recommended von den Keras-Entwicklern ist.