boruta_py icon indicating copy to clipboard operation
boruta_py copied to clipboard

max_iter values

Open sacmax opened this issue 3 years ago • 3 comments

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

sacmax avatar Apr 15 '22 14:04 sacmax

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.

Wuuzzaa avatar Apr 15 '22 15:04 Wuuzzaa

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

sacmax avatar Apr 15 '22 15:04 sacmax

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.

Wuuzzaa avatar Apr 15 '22 16:04 Wuuzzaa