Skip to content

Instantly share code, notes, and snippets.

@avisheknag17

avisheknag17/auto_arima_ts.py

Created September 15, 2020 12:20
Show Gist options
  • Save avisheknag17/059b89394c8b3d4237bd2aa5b8007a61 to your computer and use it in GitHub Desktop.
Save avisheknag17/059b89394c8b3d4237bd2aa5b8007a61 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
auto_arima_ts.py
from pmdarima.arima import auto_arima
auto_model = auto_arima(Y_train.to_numpy(), exogenous=X_train.to_numpy(), m=2, seasonal=True,
suppress_warnings = True,
step_wise=True, trace=True)
auto_model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment