auto.arima() equivalent for python

前端 未结 8 978
旧时难觅i
旧时难觅i 2020-11-28 03:30

I am trying to predict weekly sales using ARMA ARIMA models. I could not find a function for tuning the order(p,d,q) in statsmodels. Currentl

8条回答
  •  情书的邮戳
    2020-11-28 04:11

    actually

    def objfunc(order,*params ):    
        from statsmodels.tsa.arima_model import ARIMA   
        p,d,q = order   
        fit = ARIMA(endog, order, exog).fit()  
        return fit.aic()    
    from scipy.optimize import brute
    grid = (slice(1, 3, 1), slice(1, 3, 1), slice(1, 3, 1))
    brute(objfunc, grid, args=params, finish=None)
    

提交回复
热议问题