_SklearnAdapter

class _SklearnAdapter(regressor: sklearn.base.RegressorMixin)

Bases: etna.models.base.BaseAdapter

Inherited-members

Parameters

regressor (sklearn.base.RegressorMixin) –

Methods

fit(df, regressors)	Fit Sklearn model.
get_model()	Get internal sklearn model that is used inside etna class.
predict(df)	Compute predictions from a Sklearn model.

fit(df: pandas.core.frame.DataFrame, regressors: List[str]) → etna.models.sklearn._SklearnAdapter

Fit Sklearn model.

Parameters

• df (pandas.core.frame.DataFrame) – Features dataframe

• regressors (List[str]) – List of the columns with regressors

Returns

Fitted model

Return type

etna.models.sklearn._SklearnAdapter

get_model() → sklearn.base.RegressorMixin

Get internal sklearn model that is used inside etna class.

Returns

Internal model

Return type

sklearn.base.RegressorMixin

predict(df: pandas.core.frame.DataFrame) → numpy.ndarray

Compute predictions from a Sklearn model.

Parameters

df (pandas.core.frame.DataFrame) – Features dataframe

Returns

Array with predictions

Return type

numpy.ndarray