Adapt ablation and importance api examples

Author: Sarah Krebs

examples/api/ablation_paths.py

@@ -13,17 +13,17 @@
 
 if __name__ == "__main__":
     # Instantiate the run
-    run = DeepCAVERun.from_path(Path("logs/DeepCAVE/minimal/run_2"))
+    run = DeepCAVERun.from_path(Path("logs/DeepCAVE/minimal/run_1"))
 
     objective_id1 = run.get_objective_ids()[0]
-    objective_id2 = run.get_objective_ids()[1]
-    budget_id = run.get_budget_ids()[1]
+    objective_id2 = None  # replace with run.get_objective_ids()[1] for multi-objective importance
+    budget_id = run.get_budget_ids()[0]
 
     # Instantiate the plugin
     plugin = AblationPaths()
     inputs = plugin.generate_inputs(
         objective_id1=objective_id1,
-        objective_id2=None,  # replace with objective_id2 for multi-objective importance
+        objective_id2=objective_id2,
         budget_id=budget_id,
         n_hps=100,
         n_trees=100,
@@ -35,4 +35,5 @@
     # Finally, you can load the figure. Here, the filter variables play a role.
     figure1, figure2 = plugin.load_outputs(run, inputs, outputs)
     figure1.write_image("examples/api/ablation_paths_performance.png", scale=2.0)
-    figure2.write_image("examples/api/ablation_paths_improvement.png", scale=2.0)
+    if not objective_id2:
+        figure2.write_image("examples/api/ablation_paths_improvement.png", scale=2.0)

examples/api/importances.py

@@ -16,15 +16,15 @@
     run = DeepCAVERun.from_path(Path("logs/DeepCAVE/minimal/run_2"))
 
     objective_id1 = run.get_objective_ids()[0]
-    objective_id2 = run.get_objective_ids()[1]
+    objective_id2 = None  # replace with run.get_objective_ids()[1] for multi-objective importance
     budget_ids = run.get_budget_ids()
 
     # Instantiate the plugin
     plugin = Importances()
     inputs = plugin.generate_inputs(
         hyperparameter_names=list(run.configspace.keys()),
         objective_id1=objective_id1,
-        objective_id2=None,  # replace with objective_id2 for multi-objective importance
+        objective_id2=objective_id2,
         budget_ids=budget_ids,
         method="global",
         n_hps=3,