BestClassBinaryWrapper

class lightautoml.tasks.common_metric.BestClassBinaryWrapper(func)[source]

Bases: object

Metric wrapper to get best class prediction instead of probs.

There is cut-off for prediction by 0.5.

Parameters

func (Callable) – Metric function. Function format: func(y_pred, y_true, weights, **kwargs).