Python中Statsmodels ARIMA的多个输入

Question

我正在尝试使用多个输入来拟合ARIMA模型。 只要输入是单个数组，它就可以正常工作。

在这里 ，建议我将输入数组放入类似多维数组的结构中。 所以我做了：

import numpy as np
from statsmodels.tsa.arima_model import ARIMA

a = [1, 2, 3]
b = [4, 5, 6]

data = np.dstack([a, b])
for p in range(6):
    for d in range(2):
        for q in range(4):
            order = (p,d,q)         
            try:
                model = ARIMA(data, order=(p,d,q))
                print("this works:{}, {}, {} ".format(p,d,q))
            except:
                pass

但是，此脚本的输出是这样的：

   this works:0, 0, 0

显然，出了点问题（如果p，d，q全部为0，则根本不起作用）。 有人知道我在做什么错吗？

我们会向我指出正确的方向的任何建议。

Answer 1

使用ARIMA进行建模时，您需要具有足够的“自由度”。

因此，代码的问题在于np.dstack生成的数组形状为（1,3,2），这意味着它只有一个数据元素。 您至少需要6个数据元素才能运行p值为5的ARIMA模型。

数组操作示例 。 我使用np.vstack产生尽可能多的行。

请运行下面的代码片段，您将了解。

import numpy as np
from statsmodels.tsa.arima_model import ARIMA

a = [1, 2]
b = [3, 4]
c = [5, 6]
d = [7, 8]

data = np.vstack([a, b, c, d])

print(data.shape)
print(data)

for p in range(4):
    for d in range(1):
        for q in range(2):
            order = (p,d,q)    
            try:
                model = ARIMA(data, order=(p,d,q))
                print("this works:{}, {}, {} ".format(p,d,q))
            except:
                print(order)
                print('reached exception')
                pass