Fix varying configspace index order in PDP and symbolic explanations

automl · Jan 15, 2024 · 7777934 · 7777934
1 parent 77e2246
commit 7777934
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -12,6 +12,7 @@
 
 ## Bug-Fixes
 - Don't convert BOHB runs with status 'running' (consistent with SMAC).
+- Fix varying configspace index order in PDP and symbolic explanations.
 
 # Version 1.1.3
 

diff --git a/deepcave/plugins/hyperparameter/pdp.py b/deepcave/plugins/hyperparameter/pdp.py
@@ -190,6 +190,10 @@ def process(run, inputs):
         surrogate_model = RandomForestSurrogate(run.configspace, seed=0)
         surrogate_model.fit(X, Y)
 
+        # This is necessary as the run configspace can differ from the surrogate, which would be problematic as the
+        # surrogate configspace is used in the PDP and the run configspace for plotting in load_outputs()
+        surrogate_model.config_space = run.configspace
+
         # Prepare the hyperparameters
         selected_hyperparameters = [hp1]
         if hp2 is not None and hp2 != "":

diff --git a/deepcave/plugins/hyperparameter/symbolic_explanations.py b/deepcave/plugins/hyperparameter/symbolic_explanations.py
@@ -267,6 +267,10 @@ def process(run, inputs):
         surrogate_model = RandomForestSurrogate(run.configspace, seed=0)
         surrogate_model.fit(X, Y)
 
+        # This is necessary as the run configspace can differ from the surrogate, which would be problematic as the
+        # surrogate configspace is used in the PDP and the run configspace for plotting in load_outputs()
+        surrogate_model.config_space = run.configspace
+
         # Prepare the hyperparameters
         selected_hyperparameters = [hp1]
         idx1 = run.configspace.get_idx_by_hyperparameter_name(hp1)