Keras - 指定LSTM图层的输出形状

时间:2017-12-13 19:31:04

标签: python keras lstm dimension

我在Keras有顺序模型,输入和输出数据各有形状(10000,10,300):

model = Sequential()
model.add(LSTM(input_shape=(10,300), units=300, return_sequences=False, activation="sigmoid", kernel_initializer="glorot_normal", recurrent_initializer="glorot_normal"))
model.add(LSTM(return_sequences=True, units=300, activation="sigmoid", kernel_initializer="glorot_normal", recurrent_initializer="glorot_normal"))
model.compile(loss="cosine_proximity", optimizer="adam", metrics=["accuracy"])

我希望第一层只返回最后一个输出(层中每个单元格的结果(如果我理解正确),因为我需要整个序列的上下文)并将其作为输入提供给第二层(这就是我使用{{1在第一层)。第二层应该采用它并输出完整的形状序列(10,300)。

我收到了这个错误:

return_sequences=False

如何为第二层指定第一层的正确输出形状以接受它? 或者我应该采取不同的方式吗?

如果您需要任何进一步的信息,我会提供。

感谢您的回复

0 个答案:

没有答案