SKLoss

class lightautoml.tasks.losses.sklearn.SKLoss(loss, loss_params=None, fw_func=None, bw_func=None)[source]

Bases: lightautoml.tasks.losses.base.Loss

Loss used for scikit-learn.

Parameters
  • loss (str) – One of default loss function. Valid are: ‘logloss’, ‘mse’, ‘crossentropy’, ‘rmsle’.

  • loss_params (Optional[Dict]) – Addtional loss parameters.

  • 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.

set_callback_metric(metric, greater_is_better=None, metric_params=None, task_name=None)[source]

Callback metric setter.

Uses default callback of parent class Loss.

Parameters