LSTM 模型中未定義 Dropout_U 和 Dropout_W

Question

我正面臨這個錯誤

NameError: name 'Dropout_U' is not defined

通過創建 LSTM 模型

embed_dim = 128
lstm_out = 200
batch_size = 32

model = Sequential()
model.add(Embedding(2500, embed_dim,input_length = X.shape[1]))
model.add(Dropout(0.2))
model.add(LSTM(lstm_out))
model.add(Dropout_U(0.2))
model.add(Dropout_W(0.2))
model.add(Dense(2,activation='sigmoid'))
model.compile(loss = 'categorical_crossentropy', optimizer='adam',metrics = ['accuracy'])
print(model.summary())

你能幫我克服這個問題嗎？

Answer 1

使用的語法已經過時， Dropout_U已更改為recurrent_dropout 。 Dropout_W就是Dropout 。

如果您要更換Dropout_U與recurrent_dropout ，使你LSTM層的一部分它應該工作。 只需將Dropout_W層更改為Dropout 。

embed_dim = 128
lstm_out = 200
batch_size = 32

model = Sequential()
model.add(Embedding(2500, embed_dim,input_length = X.shape[1]))
model.add(Dropout(0.2))
model.add(LSTM(lstm_out, recurrent_dropout=0.2))
model.add(Dropout(0.2))
model.add(Dense(2,activation='sigmoid'))
model.compile(loss = 'categorical_crossentropy', optimizer='adam',metrics = ['accuracy'])
print(model.summary())