tabensemb.model.PytorchTabular._default_metric_sklearn

method

PytorchTabular._default_metric_sklearn(y_true, y_pred)

Calculate MSE loss for regression tasks and log loss for classification tasks using sklearn APIs.

Parameters:

y_true

Ground truth values.

y_pred

Predicted values.

Returns:

str

“mse” for regression tasks and “log_loss” for classification tasks.

float

MSE loss for regression tasks and log loss for classification tasks