Skip to content

Commit

Permalink
change max_leaves to a default of 2 for classification
Browse files Browse the repository at this point in the history
  • Loading branch information
paulbkoch committed Jan 14, 2025
1 parent 506e692 commit f402539
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/interpret-core/interpret/glassbox/_ebm/_ebm.py
Original file line number Diff line number Diff line change
Expand Up @@ -2815,7 +2815,7 @@ class ExplainableBoostingClassifier(ClassifierMixin, EBMModel):
effectively making it location-agnostic. This can lead to overfitting, especially
when the proportion of missing values is small.
- `'gain'`: Choose the best leaf for the missing value contribution at each boosting step, based on gain.
max_leaves : int, default=3
max_leaves : int, default=2
Maximum number of leaves allowed in each tree.
monotone_constraints: list of int, default=None
Expand Down Expand Up @@ -2978,7 +2978,7 @@ def __init__(
min_cat_samples: Optional[int] = 10,
cat_smooth: Optional[float] = 10.0,
missing: str = "separate",
max_leaves: int = 3,
max_leaves: int = 2,
monotone_constraints: Optional[Sequence[int]] = None,
objective: str = "log_loss",
# Overall
Expand Down

0 comments on commit f402539

Please sign in to comment.