Skip to content

Commit 0704913

Browse files
authored
evaluate features only after 5th iteration (#137)
1 parent 0b31265 commit 0704913

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

boruta/boruta_py.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -356,13 +356,15 @@ def _fit(self, X, y):
356356
# register which feature is more imp than the max of shadows
357357
hit_reg = self._assign_hits(hit_reg, cur_imp, imp_sha_max)
358358

359-
# based on hit_reg we check if a feature is doing better than
360-
# expected by chance
361-
dec_reg = self._do_tests(dec_reg, hit_reg, _iter)
362-
363-
# print out confirmed features
364-
if self.verbose > 0 and _iter < self.max_iter:
365-
self._print_results(dec_reg, _iter, 0)
359+
# Only test after the 5th round.
360+
if _iter > 4:
361+
# based on hit_reg we check if a feature is doing better than
362+
# expected by chance
363+
dec_reg = self._do_tests(dec_reg, hit_reg, _iter)
364+
365+
# print out confirmed features
366+
if self.verbose > 0 and _iter < self.max_iter:
367+
self._print_results(dec_reg, _iter, 0)
366368
if _iter < self.max_iter:
367369
_iter += 1
368370

0 commit comments

Comments
 (0)