Merge pull request #289 from understandable-machine-intelligence-lab/…

…bugfix-max-sens Fixed issue in max_sensitivity.py
understandable-machine-intelligence-lab · Aug 8, 2023 · 5fcce79 · 5fcce79
2 parents 3233228 + 748b24b
commit 5fcce79
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/quantus/metrics/robustness/avg_sensitivity.py b/quantus/metrics/robustness/avg_sensitivity.py
@@ -384,7 +384,7 @@ def evaluate_batch(
                     b=a_perturbed[instance_id].flatten(),
                 )
                 numerator = self.norm_numerator(a=sensitivities)
-                denominator = self.norm_denominator(a=x_batch[instance_id].flatten())
+                denominator = self.norm_denominator(a=a_batch[instance_id].flatten())
                 sensitivities_norm = numerator / denominator
                 similarities[instance_id, step_id] = sensitivities_norm
         mean_func = np.mean if self.return_nan_when_prediction_changes else np.nanmean

diff --git a/quantus/metrics/robustness/max_sensitivity.py b/quantus/metrics/robustness/max_sensitivity.py
@@ -383,7 +383,7 @@ def evaluate_batch(
                     b=a_perturbed[instance_id].flatten(),
                 )
                 numerator = self.norm_numerator(a=sensitivities)
-                denominator = self.norm_denominator(a=x_batch[instance_id].flatten())
+                denominator = self.norm_denominator(a=a_batch[instance_id].flatten())
                 sensitivities_norm = numerator / denominator
                 similarities[instance_id, step_id] = sensitivities_norm