Fix bug where the local importance plots were all identical independe…

…nt of the cost function used (#71)
automl · Feb 19, 2024 · 31fa103 · 31fa103
1 parent 0cda1d4
commit 31fa103
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepcave/evaluators/lpi.py b/deepcave/evaluators/lpi.py
@@ -60,7 +60,9 @@ def calculate(
         self.rs = np.random.RandomState(seed)
 
         # Get data
-        df = self.run.get_encoded_data(budget=budget, specific=True, include_combined_cost=True)
+        df = self.run.get_encoded_data(
+            objectives=objectives, budget=budget, specific=True, include_combined_cost=True
+        )
         X = df[self.hp_names].to_numpy()
         Y = df[COMBINED_COST_NAME].to_numpy()