SklearnPerSegmentModel

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

Bases: etna.models.base.PerSegmentModel

Class for holding per segment Sklearn model.

Create instance of SklearnPerSegmentModel 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 models that are used inside etna class.