Skip to content

Commit

Permalink
increase the default max_rounds value to 25000
Browse files Browse the repository at this point in the history
  • Loading branch information
paulbkoch committed Feb 27, 2024
1 parent 14f140c commit 9cbb353
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions python/interpret-core/interpret/glassbox/_ebm/_ebm.py
Original file line number Diff line number Diff line change
Expand Up @@ -2387,7 +2387,7 @@ class ExplainableBoostingClassifier(EBMModel, ClassifierMixin, ExplainerMixin):
Number of initial highly regularized rounds to set the basic shape of the main effect feature graphs.
interaction_smoothing_rounds : int, default=50
Number of initial highly regularized rounds to set the basic shape of the interaction effect feature graphs during fitting.
max_rounds : int, default=5000
max_rounds : int, default=25000
Total number of boosting rounds with n_terms boosting steps per round.
early_stopping_rounds : int, default=50
Number of rounds with no improvement to trigger early stopping. 0 turns off
Expand Down Expand Up @@ -2528,7 +2528,7 @@ def __init__(
refresh_rate=0.0,
smoothing_rounds: Optional[int] = 200,
interaction_smoothing_rounds: Optional[int] = 50,
max_rounds: Optional[int] = 5000,
max_rounds: Optional[int] = 25000,
early_stopping_rounds: Optional[int] = 50,
early_stopping_tolerance: Optional[float] = 0.0,
# Trees
Expand Down Expand Up @@ -2682,7 +2682,7 @@ class ExplainableBoostingRegressor(EBMModel, RegressorMixin, ExplainerMixin):
Number of initial highly regularized rounds to set the basic shape of the main effect feature graphs.
interaction_smoothing_rounds : int, default=50
Number of initial highly regularized rounds to set the basic shape of the interaction effect feature graphs during fitting.
max_rounds : int, default=5000
max_rounds : int, default=25000
Total number of boosting rounds with n_terms boosting steps per round.
early_stopping_rounds : int, default=50
Number of rounds with no improvement to trigger early stopping. 0 turns off
Expand Down Expand Up @@ -2823,7 +2823,7 @@ def __init__(
refresh_rate=0.0,
smoothing_rounds: Optional[int] = 200,
interaction_smoothing_rounds: Optional[int] = 50,
max_rounds: Optional[int] = 5000,
max_rounds: Optional[int] = 25000,
early_stopping_rounds: Optional[int] = 50,
early_stopping_tolerance: Optional[float] = 0.0,
# Trees
Expand Down

0 comments on commit 9cbb353

Please sign in to comment.