首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >LSTM模型中未定义Dropout_U和Dropout_W

LSTM模型中未定义Dropout_U和Dropout_W
EN

Stack Overflow用户
提问于 2020-10-16 10:13:13
回答 1查看 782关注 0票数 0

我正面临着这个错误

代码语言:javascript
运行
复制
NameError: name 'Dropout_U' is not defined

通过创建LSTM模型

代码语言:javascript
运行
复制
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())

你能帮我解决这个问题吗?

EN

回答 1

Stack Overflow用户

发布于 2021-05-10 15:43:53

使用的语法已过时,Dropout_U已更改为recurrent_dropoutDropout_W就是简单的Dropout

如果您将Dropout_U替换为recurrent_dropout,并使其成为您的LSTM层的一部分,那么它应该可以工作。只需将Dropout_W层更改为Dropout即可。

代码语言:javascript
运行
复制
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())
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/64382011

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档