Update utils.py

Smartappli · May 25, 2024 · e9dee88 · e9dee88
1 parent 0d9922f
commit e9dee88
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/fl_common/models/utils.py b/fl_common/models/utils.py
@@ -461,10 +461,7 @@ def generate_xai_heatmaps(model, image_tensor, label, save_dir, methods=None):
         plt.imshow(attributions_np, cmap="viridis")
         plt.title(f"XAI Heatmap for {method_name} (Label: {label})")
         plt.colorbar()
-        save_path = os.path.join(
-            save_dir,
-            f"xai_heatmap_{method_name}_{label}.png",
-        )
+        save_path = save_dir, + '/' + f"xai_heatmap_{method_name}_{label}.png"
         plt.savefig(save_path)
         plt.show()