@@ -3086,12 +3086,13 @@ class ExplainableBoostingClassifier(ClassifierMixin, EBMModel):
30863086 amount of overfitting of the individual models can improve the accuracy of
30873087 the ensemble as a whole.
30883088 callback : Optional[Callable[[int, int, bool, float], bool]], default=None
3089- A user-defined function that is invoked at the end of each boosting step to determine
3089+ A user-defined function that is invoked at the end of each boosting step attempt to determine
30903090 whether to terminate boosting or continue. If it returns True, the boosting loop is
30913091 stopped immediately. By default, no callback is used and training proceeds according
30923092 to the early stopping settings. The callback function receives:
3093- (1) the bag index, (2) the number of boosting steps completed,
3094- (3) a boolean indicating whether progress was made in the current step, and (4) the current best score.
3093+ (1) the bag index, (2) the callback step index (increments on every callback, even when
3094+ no update is applied), (3) a boolean indicating whether progress was made in the current step,
3095+ and (4) the current best score.
30953096 min_samples_leaf : int, default=4
30963097 Minimum number of samples allowed in the leaves.
30973098 min_hessian : float, default=1e-4
@@ -3596,12 +3597,13 @@ class ExplainableBoostingRegressor(RegressorMixin, EBMModel):
35963597 amount of overfitting of the individual models can improve the accuracy of
35973598 the ensemble as a whole.
35983599 callback : Optional[Callable[[int, int, bool, float], bool]], default=None
3599- A user-defined function that is invoked at the end of each boosting step to determine
3600+ A user-defined function that is invoked at the end of each boosting step attempt to determine
36003601 whether to terminate boosting or continue. If it returns True, the boosting loop is
36013602 stopped immediately. By default, no callback is used and training proceeds according
36023603 to the early stopping settings. The callback function receives:
3603- (1) the bag index, (2) the number of boosting steps completed,
3604- (3) a boolean indicating whether progress was made in the current step, and (4) the current best score.
3604+ (1) the bag index, (2) the callback step index (increments on every callback, even when
3605+ no update is applied), (3) a boolean indicating whether progress was made in the current step,
3606+ and (4) the current best score.
36053607 min_samples_leaf : int, default=4
36063608 Minimum number of samples allowed in the leaves.
36073609 min_hessian : float, default=0.0
0 commit comments