File tree Expand file tree Collapse file tree
python/interpret-core/interpret/glassbox/_ebm Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments