欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python 时间序列预测 ——滑动窗口法

程序员文章站 2022-06-09 14:40:02
...

有些地方的英文称之为 moving average,但这里讲的不是 ARIMA 模型中的 MA,而只是一个简单的滑动窗口内平均。

代码片段:用前30个历史值得平均来预测下一时刻

predictions = []
for i in range(len(val_dataset.columns)):
    if i == 0:
        predictions.append(np.mean(train_dataset[train_dataset.columns[-30:]].values, axis=1))
    if i < 31 and i > 0:
        predictions.append(0.5 * (np.mean(train_dataset[train_dataset.columns[-30+i:]].values, axis=1) + \
                                  np.mean(predictions[:i], axis=0)))
    if i > 31:
        predictions.append(np.mean([predictions[:i]], axis=1))
    
predictions = np.transpose(np.array([row.tolist() for row in predictions]))

python 时间序列预测 ——滑动窗口法

相关标签: 时间序列