数据中LSTM的输入形状是什么？

Question

我有一个用户位置数据集，需要根据先前的数据预测用户的下一个位置。我已选择使用lstm进行预测，但仍会影响数据的形状。

lstm的数据输入形状应该是什么？

Total number of rows: 53398560
Total number of columns: 4

Columns: User, Timestamp, slot, Location

Users count: 111247
slot : [0-95] Each slot is the 15 mins of the timestamp.
     2014-01-18 00:00:00 ----- 0
     2014-01-18 23:45:00 ----- 95
No of timestamps for each user = 480 (96 slots * 5 days)

Location: Random points for each user

上面的数据是什么Xtrain，Ytrain，Xtest，Ytest。 我需要根据480个时间戳来预测每个用户的下一个位置。

Answer 1

keras lstm层输入是一个3D张量，其形状为（batch_size，timesteps，input_dim）。从此处的文档https://keras.io/layers/recurrent/

因此，在您的情况下，应为[batch_size，480、2]（如果lstm的唯一尺寸是'slot'和'location'变量。