tabensemb.model.AbstractModel._default_metric_sklearn#

method

AbstractModel._default_metric_sklearn(y_true, y_pred)[source]#

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