feature_selector¶

Module Contents¶

Classes Summary¶

FeatureSelector

Selects top features based on importance weights.

Contents¶

class evalml.pipelines.components.transformers.feature_selection.feature_selector.FeatureSelector(parameters=None, component_obj=None, random_seed=0, **kwargs)[source]¶

Selects top features based on importance weights.

Parameters

parameters (dict) – Dictionary of parameters for the component. Defaults to None.
component_obj (obj) – Third-party objects useful in component implementation. Defaults to None.
random_seed (int) – Seed for the random number generator. Defaults to 0.

Attributes

model_family	ModelFamily.NONE
modifies_features	True
modifies_target	False

Methods

`clone`	Constructs a new component with the same parameters and random state.
`default_parameters`	Returns the default parameters for this component.
`describe`	Describe a component and its parameters
`fit`	Fits component to data
`fit_transform`	Fits on X and transforms X
`get_names`	Get names of selected features.
`load`	Loads component at file path
`name`	Returns string name of this component
`needs_fitting`	Returns boolean determining if component needs fitting before
`parameters`	Returns the parameters which were used to initialize the component
`save`	Saves component at file path
`transform`	Transforms input data by selecting features. If the component_obj does not have a transform method, will raise an MethodPropertyNotFoundError exception.

clone(self)¶

Constructs a new component with the same parameters and random state.

Returns: A new instance of this component with identical parameters and random state.

default_parameters(cls)¶

Returns the default parameters for this component.

Our convention is that Component.default_parameters == Component().parameters.

Returns: default parameters for this component.
Return type: dict

describe(self, print_name=False, return_dict=False)¶

Describe a component and its parameters

Parameters

print_name (bool, optional) – whether to print name of component
return_dict (bool, optional) – whether to return description as dictionary in the format {“name”: name, “parameters”: parameters}

Returns

prints and returns dictionary

Return type

None or dict

fit(self, X, y=None)¶

Fits component to data

Parameters

X (list, pd.DataFrame or np.ndarray) – The input training data of shape [n_samples, n_features]
y (list, pd.Series, np.ndarray, optional) – The target training data of length [n_samples]

Returns

self

fit_transform(self, X, y=None)[source]¶

Fits on X and transforms X

Parameters

X (pd.DataFrame) – Data to fit and transform
y (pd.Series) – Target data

Returns

Transformed X

Return type

pd.DataFrame

get_names(self)[source]¶

Get names of selected features.

Returns: List of the names of features selected
Return type: list[str]

static load(file_path)¶

Loads component at file path

Parameters: file_path (str) – Location to load file
Returns: ComponentBase object

property name(cls)¶: Returns string name of this component

needs_fitting(self)¶: Returns boolean determining if component needs fitting before calling predict, predict_proba, transform, or feature_importances. This can be overridden to False for components that do not need to be fit or whose fit methods do nothing.

property parameters(self)¶: Returns the parameters which were used to initialize the component

save(self, file_path, pickle_protocol=cloudpickle.DEFAULT_PROTOCOL)¶

Saves component at file path

Parameters

file_path (str) – Location to save file
pickle_protocol (int) – The pickle data stream format.

Returns

None

transform(self, X, y=None)[source]¶

Transforms input data by selecting features. If the component_obj does not have a transform method, will raise an MethodPropertyNotFoundError exception.

Parameters

X (pd.DataFrame) – Data to transform.
y (pd.Series, optional) – Target data. Ignored.

Returns

Transformed X

Return type

pd.DataFrame

feature_selection rf_classifier_feature_selector