MeanSegmentEncoderTransform

class MeanSegmentEncoderTransform

Bases: etna.transforms.base.Transform, etna.transforms.base.FutureMixin

Makes expanding mean target encoding of the segment. Creates column ‘segment_mean’.

Inherited-members

Methods

fit(df)	Fit encoder.
fit_transform(df)	May be reimplemented.
inverse_transform(df)	Inverse transforms dataframe.
load(path)	Load an object.
save(path)	Save the object.
to_dict()	Collect all information about etna object in dict.
transform(df)	Get encoded values for the segment.

Attributes

idx

fit(df: pandas.core.frame.DataFrame) → etna.transforms.encoders.mean_segment_encoder.MeanSegmentEncoderTransform

Fit encoder.

Parameters

df (pandas.core.frame.DataFrame) – dataframe with data to fit expanding mean target encoder.

Returns

Fitted transform

Return type

etna.transforms.encoders.mean_segment_encoder.MeanSegmentEncoderTransform

transform(df: pandas.core.frame.DataFrame) → pandas.core.frame.DataFrame

Get encoded values for the segment.

Parameters

df (pandas.core.frame.DataFrame) – dataframe with data to transform.

Returns

result dataframe

Return type

pandas.core.frame.DataFrame