Ошибка сегментации при определении пользовательских потерь в lightgbm - PullRequest
0 голосов
/ 07 февраля 2020

Я пробовал следующую нестандартную потерю для задачи регрессии в Lightgbm и повышения type = "goss", и я получаю ошибку сегментации:

def hm(y_true, y_pred):
    residual = (y_true - y_pred).astype("float")
    residual_abs = np.fabs(residual)
    grad = np.where(residual_abs<=300, 
                    np.copysign(1, -residual)* 0.5,
                    np.where(residual_abs>=900, 
                             np.copysign(1, -residual)* 1,
                             np.copysign(1, -residual)* 2))
    hess = np.ones(y_true.shape[0]) 
    return grad, hess

Любые идеи, почему это может быть неудачу? Я использую goss и гиперпараметры по умолчанию

...