Skip to content

Commit d9288f0

Browse files
committed
ENH micro-optimize gradient boosting
1 parent 3c92686 commit d9288f0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/ensemble/gradient_boosting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -389,7 +389,7 @@ def _update_terminal_region(self, tree, terminal_regions, leaf, X, y,
389389
y = y.take(terminal_region, axis=0)
390390

391391
numerator = residual.sum()
392-
denominator = np.sum((y - residual) * (1 - y + residual))
392+
denominator = np.dot(y - residual, 1 - y + residual)
393393

394394
if denominator == 0.0:
395395
tree.value[leaf, 0, 0] = 0.0

0 commit comments

Comments
 (0)