Bug fixing

automl · Aug 29, 2024 · 46cc7c4 · 46cc7c4
1 parent 2ceab26
commit 46cc7c4
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/deepcave/evaluators/mo_ablation.py b/deepcave/evaluators/mo_ablation.py
@@ -159,6 +159,9 @@ def calculate(
             df[normed] = (df[obj.name] - df[obj.name].min()) / (
                 df[obj.name].max() - df[obj.name].min()
             )
+
+            if obj.optimize == "upper":
+                df[normed] = 1-df[normed]
             objectives_normed.append(normed)
 
             # train one model per objective

diff --git a/deepcave/evaluators/mo_fanova.py b/deepcave/evaluators/mo_fanova.py
@@ -100,6 +100,8 @@ def calculate(
             df[normed] = (df[obj.name] - df[obj.name].min()) / (
                 df[obj.name].max() - df[obj.name].min()
             )
+            if obj.optimize == "upper":
+                df[normed] = 1-df[normed]
             objectives_normed.append(normed)
         df = df.dropna(subset=objectives_normed)
         X = df[self.hp_names].to_numpy()

diff --git a/deepcave/evaluators/mo_lpi.py b/deepcave/evaluators/mo_lpi.py
@@ -127,6 +127,8 @@ def calculate(
             df[normed] = (df[obj.name] - df[obj.name].min()) / (
                 df[obj.name].max() - df[obj.name].min()
             )
+            if obj.optimize == "upper":
+                df[normed] = 1-df[normed]
             objectives_normed.append(normed)
         df = df.dropna(subset=objectives_normed)
         X = df[self.hp_names].to_numpy()