BestClassMulticlassWrapper

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

Bases: object

Metric wrapper to get best class prediction instead of probs for multiclass.

Prediction provides by argmax.

__init__(func)[source]
Parameters

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