[ENH] Add prediction intervals to ARIMA iterative_forecast (#3100) #3106
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
[ENH] Add prediction intervals to ARIMA iterative_forecast (#3100)
Reference Issues/PRs
What does this implement/fix?
This PR adds optional prediction interval support to the
ARIMA.iterative_forecastmethod inaeon.forecasting.stats._arima.Key features:
alpha(default: None)alpha=None→ returns the samenp.ndarrayas before.alphais provided → returns apandas.DataFramewith columns:"mean"- point forecasts"lower"- lower prediction interval"upper"- upper prediction intervalscipy.stats.norm.ppfsqrt(horizon)Additional updates:
AutoARIMA.iterative_forecastnow forwards thealphaargument to the wrapped ARIMA model.This enhancement provides users with basic forecast uncertainty estimates while keeping the ARIMA implementation lightweight and backward compatible.
Does this contribution introduce a new dependency?
No.
Uses existing NumPy, SciPy, and pandas packages already in the project.
Any other comments?
ruff,flake8,isort,black, etc.).PR checklist
For all contributions
[ENH].For new estimators/functions
For developers with write access