diff --git a/supervised/base_automl.py b/supervised/base_automl.py index e91052d8..f952c7d5 100644 --- a/supervised/base_automl.py +++ b/supervised/base_automl.py @@ -1351,7 +1351,7 @@ def _validate_eval_metric(self): Use 'logloss'" ) - elif self._get_ml_task() == REGRESSION and self.eval_metric != "rmse": + elif self._get_ml_task() == REGRESSION and self.eval_metric not in ["rmse", "mse", "mae"]: raise ValueError( f"Metric {self.eval_metric} is not allowed in ML task: {self._get_ml_task()}. \ Use 'rmse'" diff --git a/supervised/utils/metric.py b/supervised/utils/metric.py index f4047099..16187f39 100644 --- a/supervised/utils/metric.py +++ b/supervised/utils/metric.py @@ -45,11 +45,10 @@ def __init__(self, params): raise MetricException("Metric name not defined") self.minimize_direction = self.name in [ "logloss", + "auc", # negative auc "rmse", "mae", - "ce", "mse", - "auc", ] if self.name == "logloss": self.metric = logloss