lgb_f1_loss_multiclass

lightautoml.tasks.losses.lgb_custom.lgb_f1_loss_multiclass(preds, train_data, clip=1e-05)[source]

Custom loss for optimizing f1.

Parameters:

  • preds (ndarray) – Predctions.

  • train_data (Dataset) – Dataset in LightGBM format.

  • clip (float) – Clump constant.

Return type:

Tuple[ndarray, ndarray]

Returns:

Gradient, hessian.