Skip to content

Commit f9cdd44

Browse files
authored
fixed count for rejected features in final iteration (#71)
thanks!
1 parent f39c077 commit f9cdd44

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

boruta/boruta_py.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -566,6 +566,7 @@ def _print_results(self, dec_reg, _iter, flag):
566566
# Boruta finished running and tentatives have been filtered
567567
else:
568568
n_tentative = np.sum(self.support_weak_)
569+
n_rejected = np.sum(~(self.support_|self.support_weak_))
569570
content = map(str, [n_iter, n_confirmed, n_tentative, n_rejected])
570571
result = '\n'.join([x[0] + '\t' + x[1] for x in zip(cols, content)])
571572
output = "\n\nBorutaPy finished running.\n\n" + result

0 commit comments

Comments
 (0)