Statsmodel ARIMA多输入

时间:2018-03-12 12:40:42

标签: python statsmodels arima

我想创建我的第一个(季节性)ARIMA模型,但我发现Statsmodel ARIMA documentation不足。我缺乏有关从多个阵列计算预测的信息(这些是numpy数组)。这些numpy数组是一天中每分钟的一系列值。我想使用去年每一天的数据进行预测。

关于如何做的任何建议/建议/链接/提示?

我正在使用Python 3.6。

1 个答案:

答案 0 :(得分:2)

您需要将数组放入一个类似多维数组的结构(Pandas DataFrame或NumPy数组)。假设您有两个数组a = [1, 2, 3]b = [4, 5, 6]

data = np.dstack([a, b])
model = statsmodels.tsa.arima_model.ARIMA(data, order=(5,1,0)) # fits ARIMA(5,1,0) model

有关创建ARIMA模型的更全面示例,请参阅this blog post