tabensemb.data.dataprocessor.FeatureValueSelector

class tabensemb.data.dataprocessor.FeatureValueSelector(**kwargs)

Bases: AbstractProcessor

Select data with the specified feature value.

Parameters:

feature: str

The feature that will be filtered.

value: float

The selected feature value.

Notes

The FeatureValueSelector will not change anything in the upcoming dataset, which means that the value in the upcoming set may exceed the range you expect. A typical error can be “IndexError: index out of range in self” from torch.embedding because of categorical features.

Methods

__init__(**kwargs)

_fit_transform(data, datamodule)
_required_kwargs()	kwargs required by the class.
_transform(data, datamodule)