From bf9cf2bd61ab31c2c8e026411f427090a879891b Mon Sep 17 00:00:00 2001 From: "Wieczorek, Alexander" Date: Mon, 13 Jan 2025 20:48:30 +0100 Subject: [PATCH] Filter data to measurement parameters only when initializing from recommender --- baybe/insights/shap.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/baybe/insights/shap.py b/baybe/insights/shap.py index c30385862..d087d53d6 100644 --- a/baybe/insights/shap.py +++ b/baybe/insights/shap.py @@ -224,9 +224,11 @@ def from_recommender( searchspace, objective, measurements ) + data = measurements[[p.name for p in searchspace.parameters]] + return cls.from_surrogate( surrogate_model, - searchspace.transform(measurements) if use_comp_rep else measurements, + searchspace.transform(data) if use_comp_rep else data, explainer_cls=explainer_cls, use_comp_rep=use_comp_rep, )