Keras指定LSTM层的输出形状

2024-06-02 05:48:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我在Keras中有顺序模型,其中输入和输出数据都有形状(10000、10300):

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"])

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

我收到这个错误:

^{pr2}$

如何指定第一层的正确输出形状,以便第二层接受它? 或者我应该用不同的方式来做吗?在

如果你需要任何进一步的信息,我会提供给他们。在

谢谢你的回复


Tags: addfalsemodelreturnactivationkernel形状units