Skip to content

Commit f402539

Browse files
committed
change max_leaves to a default of 2 for classification
1 parent 506e692 commit f402539

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • python/interpret-core/interpret/glassbox/_ebm

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2815,7 +2815,7 @@ class ExplainableBoostingClassifier(ClassifierMixin, EBMModel):
28152815
effectively making it location-agnostic. This can lead to overfitting, especially
28162816
when the proportion of missing values is small.
28172817
- `'gain'`: Choose the best leaf for the missing value contribution at each boosting step, based on gain.
2818-
max_leaves : int, default=3
2818+
max_leaves : int, default=2
28192819
Maximum number of leaves allowed in each tree.
28202820
monotone_constraints: list of int, default=None
28212821
@@ -2978,7 +2978,7 @@ def __init__(
29782978
min_cat_samples: Optional[int] = 10,
29792979
cat_smooth: Optional[float] = 10.0,
29802980
missing: str = "separate",
2981-
max_leaves: int = 3,
2981+
max_leaves: int = 2,
29822982
monotone_constraints: Optional[Sequence[int]] = None,
29832983
objective: str = "log_loss",
29842984
# Overall

0 commit comments

Comments
 (0)