diff --git a/sklearn/ensemble/gradient_boosting.py b/sklearn/ensemble/gradient_boosting.py index 4a6f4133033e9..295c6ead6c32c 100644 --- a/sklearn/ensemble/gradient_boosting.py +++ b/sklearn/ensemble/gradient_boosting.py @@ -421,10 +421,10 @@ def __call__(self, y, pred, sample_weight=None): mask = y > pred if sample_weight is None: loss = (alpha * diff[mask].sum() + - (1.0 - alpha) * diff[~mask].sum()) / y.shape[0] + (1.0 - alpha) * (-diff[~mask]).sum()) / y.shape[0] else: loss = ((alpha * np.sum(sample_weight[mask] * diff[mask]) + - (1.0 - alpha) * np.sum(sample_weight[~mask] * diff[~mask])) / + (1.0 - alpha) * np.sum(sample_weight[~mask] * (-diff[~mask]))) / sample_weight.sum()) return loss