NonPredictionIntervalContextRequiredAbstractModel

class NonPredictionIntervalContextRequiredAbstractModel[source]

Bases: etna.models.base.AbstractModel

Interface for models that don’t support prediction intervals and need context for prediction.

Inherited-members

Methods

fit(ts)

Fit model.

forecast(ts, prediction_size)

Make predictions.

get_model()

Get internal model/models that are used inside etna class.

load(path)

Load an object.

predict(ts, prediction_size)

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.
abstract forecast(ts: etna.datasets.tsdataset.TSDataset, prediction_size: int) → etna.datasets.tsdataset.TSDataset[source]

Make predictions.

Parameters

  • ts (etna.datasets.tsdataset.TSDataset) – Dataset with features

  • prediction_size (int) – Number of last timestamps to leave after making prediction. Previous timestamps will be used as a context for models that require it.

Returns

Dataset with predictions

Return type

etna.datasets.tsdataset.TSDataset

abstract predict(ts: etna.datasets.tsdataset.TSDataset, prediction_size: int) → etna.datasets.tsdataset.TSDataset[source]

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

Parameters

  • ts (etna.datasets.tsdataset.TSDataset) – Dataset with features

  • prediction_size (int) – Number of last timestamps to leave after making prediction. Previous timestamps will be used as a context for models that require it.

Returns

Dataset with predictions

Return type

etna.datasets.tsdataset.TSDataset