LGBLoss
- class lightautoml.tasks.losses.lgb.LGBLoss(loss, loss_params=None, fw_func=None, bw_func=None)[source]
Bases:
lightautoml.tasks.losses.base.Loss
Loss used for LightGBM.
- __init__(loss, loss_params=None, fw_func=None, bw_func=None)[source]
- Parameters
loss_params (
Optional
[Dict
]) – additional loss parameters. Format like inlightautoml.tasks.custom_metrics
.fw_func (
Optional
[Callable
]) – forward transformation. Used for transformation of target and item weights.bw_func (
Optional
[Callable
]) – backward transformation. Used for predict values transformation.
Note
Loss can be one of the types:
Str: one of default losses (‘auc’, ‘mse’, ‘mae’, ‘logloss’, ‘accuray’, ‘r2’, ‘rmsle’, ‘mape’, ‘quantile’, ‘huber’, ‘fair’) or another lightgbm objective.
Callable: custom lightgbm style objective.