Dropout训练参数
Dropout训练参数
提问于 2020-11-06 09:00:20
我希望在LSTM层中使用dropout进行训练和测试。根据这篇文章：
https://towardsdatascience.com/learning-note-dropout-in-recurrent-networks-part-2-f209222481f8
model = Sequential()
model.add(LSTM(X_len, return_sequences = True, input_shape=(X_len, 1)))
model.add(Dropout(rate=0.2, training=True))
model.add(LSTM(X_len))
model.add(Dropout(rate=0.2, training=True))
model.add(Dense(Y_len))我得到的错误是：
TypeError: ('Keyword argument not understood:', 'training')你知道怎么解决这个问题吗？
回答 1
Stack Overflow用户
回答已采纳
发布于 2020-11-06 16:16:03
使用函数格式
inp = Input(shape=(X_len, 1))
x = LSTM(X_len, return_sequences = True)(inp)
x = Dropout(rate=0.2)(x, training=True)
x = LSTM(X_len)(x)
x = Dropout(rate=0.2)(x, training=True)
out = Dense(Y_len)(x)
model = Model(inp, out)
model.compile('adam', 'mse')
model.summary()页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/64707451
复制相关文章
相似问题
腾讯云开发者
