add more regression metrics (#251)

mljar · Nov 27, 2020 · 12ba315 · 12ba315
1 parent c2d8ccd
commit 12ba315
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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