Skip to content

Commit 0a76e21

Browse files
committed
avoid detecting interactions when boosting exits early
1 parent c46a34d commit 0a76e21

File tree

1 file changed

+3
-0
lines changed
  • python/interpret-core/interpret/glassbox/_ebm

1 file changed

+3
-0
lines changed

python/interpret-core/interpret/glassbox/_ebm/_ebm.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1170,6 +1170,9 @@ def fit(self, X, y, sample_weight=None, bags=None, init_score=None):
11701170
rngs.append(bagged_rng)
11711171

11721172
while True: # this isn't for looping. Just for break statements to exit
1173+
if stop_flag.value:
1174+
break
1175+
11731176
if interactions is None:
11741177
break
11751178

0 commit comments

Comments
 (0)