最小的例子:
from statsforecast import StatsForecast
from statsforecast.models import AutoARIMA
import pandas as pd
df = pd.read_csv('https://datasets-nixtla.s3.amazonaws.com/air-passengers.csv')
sf = StatsForecast(
models = [AutoARIMA(season_length = 12)],
freq = 'M',
n_jobs=-1,
verbose=True
)
sf.fit(df)
如何获取拟合模型的参数?
我知道使用
pmdarima
包是可能的,但是 pmdarima 太慢了并且在大数据上耗尽了内存。 statsforecast
似乎很有希望,但前提是有办法获得参数
从this answer,解决方案是:
sf.fitted_[0][0].model_['arma']
这将输出一个包含 7 个值的元组。我不知道参数到元组值的确切映射,但从这一行看来是:
(p, d, q, P, D, Q, constant)