keras - keras 中的多对一实现

Question

我正在尝试实现多对一模型，其中输入是大小为 的词向量d .我需要输出一个大小为 的向量d 在 LSTM 结束时。



在此 question ，提到使用(对于多对一模型)

model = Sequential()
model.add(LSTM(1, input_shape=(timesteps, data_dim)))

我仍然怀疑如何在 keras 模型中指定输入，更重要的是输出维度

Answer 1

您提供的代码实际上是针对 one-to-many 的场景 - 但对于大小为 1 的输出。如果您想要输出大小为 d您需要尝试以下代码:

model = Sequential()
model.add(LSTM(d, input_shape=(len_seq, d)))

此外 - 对于词向量，我强烈建议您使用 Embedding 专为您的用例设计的层。将其用作 LSTM 的输入然后。