MultiSegmentModel

class MultiSegmentModel(base_model: Any)

Bases: etna.models.base.FitAbstractModel, etna.models.base.ForecastAbstractModel, etna.core.mixins.BaseMixin

Class for holding specific models for per-segment prediction.

Init MultiSegmentModel.

Parameters

base_model (Any) – Internal model which will be used to forecast segments, expected to have fit/predict interface

Inherited-members

Methods

fit(ts)	Fit model.
forecast(ts)	Make predictions.
get_model()	Get internal model that is used inside etna class.

fit(ts: etna.datasets.tsdataset.TSDataset) → etna.models.base.MultiSegmentModel

Fit model.

Parameters

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

Returns

Model after fit

Return type

etna.models.base.MultiSegmentModel

forecast(ts: etna.datasets.tsdataset.TSDataset) → etna.datasets.tsdataset.TSDataset

Make predictions.

Parameters

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

Returns

Dataset with predictions

Return type

etna.datasets.tsdataset.TSDataset

get_model() → Any

Get internal model that is used inside etna class.

Internal model is a model that is used inside etna to forecast segments, e.g. catboost.CatBoostRegressor or sklearn.linear_model.Ridge.

Returns

Internal model

Return type

Any