scikit-learn-contrib / boruta_py

Python implementations of the Boruta all-relevant feature selection method.
BSD 3-Clause "New" or "Revised" License
1.46k stars 252 forks source link

Tuple Index Out of range #47

Open freshnemo opened 5 years ago

freshnemo commented 5 years ago

Hi, my working environment is python 3.6.4 and boruta is the current version. When I runed the example dataset in the example folder of boruta. The following error message will show "tuple index out of range". It seems to be some errors happened in this line "if not_selected.shape[0] > 0 and not_selected.shape[1] > 0:"

guitarmind commented 5 years ago

Hi @freshnemo,

Thanks for reporting this! I have made a PR to fix this bug. The example notebook in boruta should also work fine as well!

Please try it out later once it is merged, thank you!