EnsembleModelFeatureSelector

class mastml.legos.feature_selectors.EnsembleModelFeatureSelector(estimator, k_features)

Bases: object

Class custom-written for MAST-ML to conduct selection of features with ensemble model feature importances

Args:

estimator: (scikit-learn model/estimator object), a scikit-learn model/estimator

k_features: (int), the number of features to select

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:

dataframe: (dataframe), dataframe of selected X features

Methods Summary

fit(X[, y])
transform(X)

Methods Documentation

fit(X, y=None)

transform(X)