From 9b912768067124f2cfcd4658bea78c9525a3e278 Mon Sep 17 00:00:00 2001 From: Christian Lorentzen Date: Sun, 18 Feb 2024 15:58:51 +0100 Subject: [PATCH] ENH minor improvement of binomial hessian --- sklearn/_loss/_loss.pyx.tp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sklearn/_loss/_loss.pyx.tp b/sklearn/_loss/_loss.pyx.tp index 61043162ae51a..cdfea45058bb2 100644 --- a/sklearn/_loss/_loss.pyx.tp +++ b/sklearn/_loss/_loss.pyx.tp @@ -765,9 +765,8 @@ cdef inline double_pair cgrad_hess_half_binomial( gh.val1 = ((1 - y_true) - y_true * gh.val2) / (1 + gh.val2) # gradient gh.val2 = gh.val2 / (1 + gh.val2)**2 # hessian else: - gh.val2 = exp(raw_prediction) + gh.val2 = exp(raw_prediction) # = 1. order Taylor in exp(raw_prediction) gh.val1 = gh.val2 - y_true - gh.val2 *= (1 - gh.val2) return gh