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.

Parameters

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