2016-07-18 57 views
5

Ho provato quanto segue:Come è possibile visualizzare un'architettura di rete neurale con Keras?

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

Tuttavia, questo dà un errore:

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' 

Come posso risolvere questo problema?

Nota: il file hdf5 e YAML è reperibile on Github.

+0

Vedere anche: [Come si visualizzano le architetture di rete neurali?] (Http://datascience.stackexchange.com/q/12851/8820) –

risposta

1

Se non si è già installato il pacchetto python pydot, provare ad installarlo. Se si dispone della reinstallazione pydot, si dovrebbe aiutare con il problema.

+1

'pydot' è installato nella versione 1.2.2. La reinstallazione non ha aiutato. –

2

Il problema viene anche fatto riferimento allo issues page del progetto keras. È necessario installare una versione di pydot < = 1.1.0 perché la funzione find_graphviz era removed nella versione 1.2.0. In alternativa è possibile installare pydot-ng invece, che è recommended dagli sviluppatori keras.

Problemi correlati