使用 statsmodels.tsa 返回训练集预测值

Posted 2023-03-25

【中文标题】使用 statsmodels.tsa 返回训练集预测值

【英文标题】：Returning training set predicted values with statsmodels.tsa

【发布时间】：2021-11-22 11:04:36

【问题描述】：

我一直在使用 statsmodels (v0.12.2) 包训练 ARIMA 模型，并想看看模型如何适应训练数据

当前代码：

from statsmodels.tsa.arima.model import ARIMA

#for some p,d,q
model = ARIMA(train, order = (p,d,q)
model_fit = model.fit()

尝试做：

我想根据实际训练值绘制训练集的预测。 阅读this documentation后，我一直在尝试使用以下内容：

model_fit.get_prediction()

但是，这会返回：

<statsmodels.tsa.statespace.mlemodel.PredictionResultsWrapper at 0x7f804bf5bbe0>

问题：

如何为训练集返回这些预测值？

建议！

【参考方案1】：

我认为您正在寻找模型的拟合值，如果是，请使用，

model_fit.fittedvalues

你可以找到一个完整的例子here。

【参考方案2】：

我发现这种变化：

model_fit.get_prediction()

到

model_fit.get_prediction().predicted_mean

返回一个不完美但适合我分析的数组。

如果您有替代/更好的方法，请发布答案！