SklearnMultiSegmentModel

class SklearnMultiSegmentModel(regressor: sklearn.base.RegressorMixin)[source]

Bases: etna.models.mixins.MultiSegmentModelMixin, etna.models.mixins.NonPredictionIntervalContextIgnorantModelMixin, etna.models.base.NonPredictionIntervalContextIgnorantAbstractModel

Class for holding Sklearn model for all segments.

Create instance of SklearnMultiSegmentModel with given parameters.

Parameters

regressor (sklearn.base.RegressorMixin) – Sklearn model for regression

Inherited-members

Methods

fit(ts)

Fit model.

forecast(ts)

Make predictions.

get_model()

Get internal model that is used inside etna class.

load(path)

Load an object.

predict(ts)

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.