Skip to content

Commit

Permalink
[Hoxfix] fix huber objective bugs (#22)
Browse files Browse the repository at this point in the history
  • Loading branch information
RektPunk authored Aug 29, 2024
1 parent 5c2e869 commit 37e776a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mqboost/objective.py
Original file line number Diff line number Diff line change
@@ -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

Expand All @@ -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:
Expand Down

0 comments on commit 37e776a

Please sign in to comment.