From c02c5df1672c1f22d664626d49322b97ad0fb8e5 Mon Sep 17 00:00:00 2001 From: RektPunk Date: Thu, 29 Aug 2024 15:58:43 +0900 Subject: [PATCH] fis juber objective bugs --- mqboost/objective.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mqboost/objective.py b/mqboost/objective.py index 3de5c5f..9185764 100644 --- a/mqboost/objective.py +++ b/mqboost/objective.py @@ -1,5 +1,5 @@ from functools import partial -from typing import Any, Callable, List, Tuple +from typing import Any, Callable, List, Optional, Tuple import numpy as np @@ -21,7 +21,7 @@ def _grad_rho(u: np.ndarray, alpha: float) -> np.ndarray: return (u < 0).astype(int) - alpha -def _hess_rho(u: np.ndarray, alpha: float) -> np.ndarray: +def _hess_rho(u: np.ndarray, alpha: float, delta: Optional[float] = None) -> np.ndarray: """ Compute the Hessian of the check and huber loss function. Args: