AutoARIMAModel

class AutoARIMAModel(**kwargs)[source]

Bases: etna.models.mixins.PerSegmentModelMixin, etna.models.mixins.PredictionIntervalContextIgnorantModelMixin, etna.models.base.PredictionIntervalContextIgnorantAbstractModel

Class for holding auto arima model.

Method predict can use true target values only on train data on future data autoregression forecasting will be made even if targets are known.

Notes

We use pmdarima.arima.arima.ARIMA.

Init auto ARIMA model with given params.

Parameters

**kwargs – Training parameters for auto_arima from pmdarima package.

Inherited-members

Methods

fit(ts)

Fit model.

forecast(ts[, prediction_interval, quantiles])

Make predictions.

get_model()

Get internal models that are used inside etna class.

load(path)

Load an object.

predict(ts[, prediction_interval, quantiles])

Make predictions with using true values as autoregression context if possible (teacher forcing).

save(path)

Save the object.

to_dict()

Collect all information about etna object in dict.

Attributes

context_size

Context size of the model.