Skip to content

Commit 4998d0b

Browse files
Merge branch 'main' of github.com:sequential-parameter-optimization/spotPython
2 parents 1d0e6ff + 8b0ec28 commit 4998d0b

2 files changed

Lines changed: 6 additions & 9 deletions

File tree

src/spotpython/fun/xai_hyperlight.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -169,15 +169,15 @@ def fun(self, X: np.ndarray, fun_control: dict = None) -> np.ndarray:
169169
# Multiply results by the weights. Positive weights mean that the result is to be minimized.
170170
# Negative weights mean that the result is to be maximized, e.g., accuracy.
171171
z_val = fun_control["weights"] * df_eval
172-
xai_incons = fun_control["xai_weight"] * xai_attr
173172

174173
# Append, since several configurations can be evaluated at once.
175174
z_res = np.append(z_res, z_val)
176-
xai_res = np.append(xai_res, xai_incons)
175+
xai_res = np.append(xai_res, xai_attr)
177176

178-
print("Performance loss: ", z_val)
179-
print("XAI inconsistency: ", xai_incons)
177+
# Combine z_res and xai_res into a single array
178+
combined_res = np.column_stack((z_res, xai_res))
180179

181-
res = z_res + xai_res
180+
# print("combined: ", combined_res)
181+
# print("shape: ", combined_res.shape)
182182

183-
return res, z_res, xai_res
183+
return combined_res

src/spotpython/light/trainmodel.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -673,23 +673,20 @@ def train_model_xai(config: dict, fun_control: dict, timestamp: bool = True) ->
673673
attribution_ig = attr_ig.attribute(X_val_tensor, baselines=baseline)
674674
ig_attr_test_sum = attribution_ig.detach().numpy().sum(0)
675675
ig_attr_test_norm_sum = ig_attr_test_sum / np.linalg.norm(ig_attr_test_sum, ord=1)
676-
print("Integrated Gradients attribution sum:", ig_attr_test_norm_sum)
677676
attributions_dict["IntegratedGradients"] = ig_attr_test_norm_sum
678677

679678
if "KernelShap" in fun_control["xai_methods"]:
680679
attr_ks = KernelShap(model)
681680
attribution_ks = attr_ks.attribute(X_val_tensor, baselines=baseline)
682681
ks_attr_test_sum = attribution_ks.detach().numpy().sum(0)
683682
ks_attr_test_norm_sum = ks_attr_test_sum / np.linalg.norm(ks_attr_test_sum, ord=1)
684-
print("KernelShap attribution sum:", ks_attr_test_norm_sum)
685683
attributions_dict["KernelShap"] = ks_attr_test_norm_sum
686684

687685
if "DeepLift" in fun_control["xai_methods"]:
688686
attr_dl = DeepLift(model)
689687
attribution_dl = attr_dl.attribute(X_val_tensor, baselines=baseline)
690688
dl_attr_test_sum = attribution_dl.detach().numpy().sum(0)
691689
dl_attr_test_norm_sum = dl_attr_test_sum / np.linalg.norm(dl_attr_test_sum, ord=1)
692-
print("DeepLift attribution sum:", dl_attr_test_norm_sum)
693690
attributions_dict["DeepLift"] = dl_attr_test_norm_sum
694691

695692
attributions_list = [attributions_dict[method] for method in fun_control["xai_methods"]]

0 commit comments

Comments
 (0)