FilterFeaturesTransform

class FilterFeaturesTransform(include: Optional[Sequence[str]] = None, exclude: Optional[Sequence[str]] = None, return_features: bool = False)[source]

Bases: etna.transforms.base.ReversibleTransform

Filters features in each segment of the dataframe.

Create instance of FilterFeaturesTransform.

Parameters
  • include (Optional[Sequence[str]]) – list of columns to pass through filter

  • exclude (Optional[Sequence[str]]) – list of columns to not pass through

  • return_features (bool) – indicates whether to return features or not.

Raises

ValueError: – if both options set or non of them

Inherited-members

Methods

fit(ts)

Fit the transform.

fit_transform(ts)

Fit and transform TSDataset.

get_regressors_info()

Return the list with regressors created by the transform.

inverse_transform(ts)

Inverse transform TSDataset.

load(path)

Load an object.

params_to_tune()

Get grid for tuning hyperparameters.

save(path)

Save the object.

set_params(**params)

Return new object instance with modified parameters.

to_dict()

Collect all information about etna object in dict.

transform(ts)

Transform TSDataset inplace.

get_regressors_info() List[str][source]

Return the list with regressors created by the transform.

Return type

List[str]