SKLoss

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

Bases: 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)

Callback metric setter.

Uses default callback of parent class Loss.

Parameters:

• metric (Union[str, Callable]) – Callback metric.

• greater_is_better (Optional[bool]) – Whether or not higher value is better.

• metric_params (Optional[Dict]) – Additional metric parameters.

• task_name (Optional[str]) – Name of task.