SklearnFeatureSelector

class mastml.feature_selectors.SklearnFeatureSelector(selector, **kwargs)

Bases: BaseSelector

Class that wraps scikit-learn feature selection methods with some new MAST-ML functionality

Args:

selector (str) : a string denoting the name of a sklearn.feature_selection object

**kwargs: the key word arguments of the designated sklearn.feature_selection object

Methods:

fit: performs feature selection

Args:

X: (dataframe), dataframe of X features

y: (dataframe), dataframe of y data

Returns:

None

transform: performs the transform to generate output of only selected features

Args:

X: (dataframe), dataframe of X features

Returns:

X_select: (dataframe), dataframe of selected X features

Methods Summary

fit(X, y)
transform(X)

Methods Documentation

fit(X, y)

transform(X)