LinearPerSegmentModel

class LinearPerSegmentModel(fit_intercept: bool = True, **kwargs)[source]

Bases: etna.models.sklearn.SklearnPerSegmentModel

Class holding per segment sklearn.linear_model.LinearRegression.

Create instance of LinearModel with given parameters.

Parameters

fit_intercept (bool) – Whether to calculate the intercept for this model. If set to False, no intercept will be used in calculations (i.e. data is expected to be centered).

Inherited-members

Methods

fit(ts)

Fit model.

forecast(ts)

Make predictions.

get_model()

Get internal models that are used inside etna class.