【发布时间】:2017-10-30 16:51:25
【问题描述】:
我需要一个模型来输出形状为(3,2) 的回归目标,其中3 是事件数,每个事件都有X and Y 坐标。输入数据可以跨越 1 个或多个时间步长。这是我的模型示例,用于输入具有一个时间步长的情况。
model = Sequential()
model.add(Embedding(vocab_size + 1, embedding, input_length=1))
model.add(LSTM(hidden, recurrent_dropout=0.1, return_sequences=True))
model.add(Flatten())
model.add(RepeatVector(3))
model.add(LSTM(2, return_sequences=True))
模型编译并运行,但它在 3 个事件中为 X, Y 生成相同的值。我认为这是因为RepeatVector 层。我还能如何确保我的输出是(None, 3, 2)?
【问题讨论】:
标签: python tensorflow keras lstm keras-layer