lstm - 一对多 LSTM

Question

我已经训练了一对多的 LSTM，并且我使用了 WandB。我创建了以下代码：

`X = list()
 Y = list()
 X = [x+3 for x in range(-2, 43, 3)]
 for i in X:
     output_vector = list()
     output_vector.append(i+1)
     output_vector.append(i+2)
     Y.append(output_vector)`

然后在 numpy 数组中重塑它们：

`X = np.array(X).reshape(15,1, 1)
 Y = np.array(Y)`

最后一对多 lstm ：

`model = Sequential()
 model.add(LSTM(50, activation='relu', input_shape=(1,1)))
 model.add(Dense(3))
 model.compile(optimizer='adam', loss='mse')
 wandb.init(project='simple-lstm')
 model.fit(X, Y, epochs=1000, validation_split=0.2, batch_size=3, 
 callbacks=[WandbCallback()])`

我想将输入和输出转换为单热向量，我该如何实现它。输入和输出向量的形状是什么？

标签： lstmone-to-many