ARIMA预测模型保存到文件中,稍后用于预测

时间:2016-04-04 06:51:16

标签: python file object model save

我想保存我为将来使用而创建的ARIMA模型对象 - 如何以最有效的形式执行此操作?现在,我创建模型,比如arima_mod,并使用arima_mod.forecast()。如何将此模型另存为文本并在以后撤消?我很确定可以使用pickle完成,但我不明白如何正确地调用它...

谢谢!

2 个答案:

答案 0 :(得分:0)

好的......一种方法是使用arima_mod.save('file'),然后按如下方式创建任意statsmodels.tsa.arima_model.ARMAResultsWrapper object

model = sm.tsa.ARIMA([0,1,2,3], order=(0, 0, 0))  
results_= model.fit()  

并使用真实的模型重写“占位符”模型

results_.load('file')

如果有人有更好的解决方案,我很乐意听到。

答案 1 :(得分:0)

(这是一个很晚的答案,但是...)

我认为您不需要创建任意结果包装器对象。只需使用类方法ARIMAResults.load()doc)。

# earlier:
some_model.save('file')

# now:
from statsmodels.tsa.arima.model import ARIMAResults
results_ = ARIMAResults.load('file')