Question

您好我正在使用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

所以我希望得到支持以克服这种情况，

Answer 1

你必须安装graphviz二进制文件，你不能用pip安装它们（而是安装一个python包装器）。使用Linux发行版的软件包管理器安装graphviz然后它应该可以工作。

如何解决以下问题，绘制keras模型？

1 个答案: