ENH/MRG: Support for pandas dataframe input.

scikit-learn-contrib / boruta_py

Python implementations of the Boruta all-relevant feature selection method.

BSD 3-Clause "New" or "Revised" License

1.5k stars 256 forks source link

ENH/MRG: Support for pandas dataframe input. #66

Closed zoj613 closed 4 years ago

zoj613 commented 4 years ago

Added support for pandas dataframe input. I think it is too restrictive of the package to only allow numpy arrays because a lot of the time people have their data in dataframe format.

Added functionality such that a dataframe input can be passed on to Boruta without converting it into a numpy array beforehand.
Added new parameter return_df on the transform method so that if the input is a data frame, one can choose to return the transformed data as a numpy array or pandas dataframe.

zoj613 commented 4 years ago

ping @danielhomola

danielhomola commented 4 years ago

I'm on holiday but looks good from a quick glance.