2017-02-13 43 views
0

你好我與keras的工作,我的代碼是罰款,直到我就開始想繪製的點如下:如何解決以下問題,繪製keras模型?

from keras.utils.visualize_util import plot 
plot(model, to_file='model.png') 

這是我的代碼的一部分:

labels = to_categorical(np.asarray(sents)) 
#Dimensions of labels and data 
showShape(data,labels) 
#Spliting data, training and testing 
x_train, y_train, x_val, y_val = createMatrix(data,labels,VALIDATION_SPLIT) 
matrixDetails(x_train, y_train, x_val, y_val) 
model = Sequential() 
model.add(Embedding(MAX_NB_WORDS, 64, dropout=0.2)) 
model.add(LSTM(64, dropout_W=0.2, dropout_U=0.2)) # try using a GRU instead, for fun 
model.add(Dense(35)) 
model.add(Activation('softmax')) 
model.compile(loss='binary_crossentropy', 
       optimizer='adam', 
metrics=['accuracy']) 
model.summary() 
train(model, x_train, y_train, 100, nb_epoch, x_val, y_val) 

save(model,'MODELS/model'+MODEL) 

from keras.utils.visualize_util import plot 
plot(model, to_file='model.png') 

,這是輸出:

e/CHECKPOINTS/checkpoint.39-0.07.hdf5 
679/679 [==============================] - 0s - loss: 0.0506 - acc: 0.9835 - val_loss: 0.0685 - val_acc: 0.9824 
Traceback (most recent call last): 
    File "lstm2.py", line 163, in <module> 
    from keras.utils.visualize_util import plot 
    File "/usr/local/lib/python3.5/dist-packages/keras/utils/visualize_util.py", line 13, in <module> 
    raise RuntimeError('Failed to import pydot. You must install pydot' 
RuntimeError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work. 

我想:

sudo pip3 install pytdot 

但是我:

Collecting pytdot 

    Could not find a version that satisfies the requirement pytdot (from versions:) 
No matching distribution found for pytdot 

我也試過:

sudo pip3 install graphviz 

和我:

Requirement already satisfied: graphviz in /usr/local/lib/python3.5/dist-packages 

所以我希望得到支持,以克服這種情況,

回答

1

你必須安裝graphviz二進制文件,你不能安裝這些點(而不是安裝一個python包裝)。使用你的Linux發行版的包管理器來安裝graphviz,然後它應該可以工作。