Skip to content

Commit 06febb1

Browse files
authored
Update boruta_py.py
1 parent b549626 commit 06febb1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

boruta/boruta_py.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -416,7 +416,7 @@ def _add_shadows_get_imps(self, X, y, dec_reg):
416416
def _assign_hits(self, hit_reg, cur_imp, imp_sha_max):
417417
# register hits for features that did better than the best of shadows
418418
cur_imp_no_nan = cur_imp[0]
419-
cur_imp_no_nan[np.isnan(cur_imp_no_nan)]=0
419+
cur_imp_no_nan[np.isnan(cur_imp_no_nan)] = 0
420420
hits = np.where(cur_imp_no_nan > imp_sha_max)[0]
421421
hit_reg[hits] += 1
422422
return hit_reg

0 commit comments

Comments
 (0)