dataframify_selector

mastml.legos.feature_selectors.dataframify_selector(transform)[source]

Method which transforms output of scikit-learn feature selectors from array to dataframe. Enables preservation of column names.

Args:

transform: (function), a scikit-learn feature selector that has a transform method

Returns:

new_transform: (function), an amended version of the transform method that returns a dataframe