SklearnMultiSegmentModel

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

Bases: etna.models.base.MultiSegmentModel

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.