search

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

max_iter values #105

Open sacmax opened 2 years ago

sacmax commented 2 years ago

Hi, Is there a way to find the optimal number for max_iter? I need to find it to reduce the runtime. thanks, Sac

Wuuzzaa commented 2 years ago

Hi,

set the early_stopping parameter = True. Take a look at n_iter_no_change too. Both of them are implementet but not in the version you get from pip (at least the last time i checked it).

Here is the part of the current source https://github.com/scikit-learn-contrib/boruta_py/blob/e04d1a17de142679eabebebd8cedde202587fbf1/boruta/boruta_py.py#L117

Hope this helps.

sacmax commented 2 years ago

Thanks Wuuzzaa for your reply. Is there a easy way to get this version either using PIP or conda?

Wuuzzaa commented 2 years ago

Conda and PIP still use version 0.3 (which does not support early stopping).

Imho the easiest way is too clone this repo (or only the boruta.py file) into your project at least this worked for me. If you do this ensure to import the cloned file and not the boruta 0.3.