如何在时间序列的LSTM模型中将未来的预测用作输入变量?

时间:2019-11-21 10:21:53

标签: machine-learning time-series lstm data-science recurrent-neural-network

我无法真正找到答案的一般问题仅暗示可能:

假设我要预测未来的销售量。

y(t+1) = sales at day t+1 (t+1 = next day)

我有两个输入变量;历史销售和历史天气预报。

x1(t) = historical sales day t
x2(t) = historical weather forecast for day t

训练模型后,我可以预测y(t + 1)。

但是,如何使用天气的未来数据作为输入?我已经有了第t + 1天的天气预报,该天气预报会影响我的销售,因此我想将其用作输入-在这种情况下为x2(t + 1)。像这样:

Output:
y(t+1)

Input:
x1(t)
x2(t)
x2(t+1) <------

是否可以在LSTM模型中合并此功能?如果是这样,则用于训练和使用模型的输入矩阵会如何?

1 个答案:

答案 0 :(得分:0)

您说得很对。您可以提供当前天气和以前的销售情况作为输入,以预测当前的销售情况。

sales[t+1] = RNN(weather[t+1], sales[t]) <-- [Correct]

但是,由于相关信息将通过隐藏功能传播,因此无需提供以前的天气。

sales[t+1] = RNN(weather[t+1], weather[t], sales[t]) <-- [Wrong]

示例

这是我们的示例数据。

df = pd.DataFrame([{'weather':1, 'sales':500}, {'weather':3, 'sales':200}, {'weather':2, 'sales':400}, {'weather':0, 'sales':600}])
print(df)

   weather  sales
0        1    500
1        3    200
2        2    400
3        0    600

我们必须生成特定尺寸的训练输入。

#Training input dimensions = (No. of training samples, seq_length, No. of features)

seq_len = 3 #Number of times the LSTM loops
n_features = 2 # weather and sales are considered as input

training_input = torch.zeros((df.shape[0], seq_len, n_features))
row = torch.zeros(seq_len, n_features)
for i in range(df.shape[0]):
    row[:-1] = row[1:]
    prev_sales = df.sales[i-1] if i > 0 else 0 #i.e., sales[-1] = 0
    row[-1, :] = torch.tensor([df.weather[i], prev_sales])
    training_input[i] = row

print(training_input)

tensor([[[  0.,   0.],
         [  0.,   0.],
         [  1.,   0.]],

        [[  0.,   0.],
         [  1.,   0.],
         [  3., 500.]],

        [[  1.,   0.],
         [  3., 500.],
         [  2., 200.]],

        [[  3., 500.],
         [  2., 200.],
         [  0., 400.]]])

以下部分是有关向LSTM层提供训练输入的示例。

初始化LSTM参数

input_size = 2 #weather and previous sales are considered as input
hidden_size = 2 #any number can be used
n_layers = 1 #number of LSTMs stacked. In this case, only 1 LSTM is used
batch_size = training_input.size()[0] #passing entire training input in one go

初始化hidden_​​input

hidden_input = torch.zeros(n_layers,batch_size,hidden_size), torch.zeros(n_layers,batch_size, hidden_size)

创建LSTM层

lstm = nn.LSTM(input_size,hidden_size)

必须根据LSTM类中正向函数的可接受输入尺寸来调整训练输入。

lstm_input = training_input.view(seq_len,batch_size,input_size)
out, hidden = lstm(lstm_input, hidden_input)

print(out[-1])

tensor([[2.0370e-10, 9.6134e-07],
    [2.2299e-25, 7.1835e-28],
    [2.0600e-10, 1.1409e-06],
    [8.0952e-21, 1.2101e-24]], grad_fn=<SelectBackward>)

有关更多信息,请参见Pytorch documentation for LSTM layer。希望这会有所帮助。