Update inverse_estimation.py

understandable-machine-intelligence-lab · Feb 23, 2024 · 6cc1cda · 6cc1cda
1 parent 1125384
commit 6cc1cda
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/quantus/metrics/inverse_estimation.py b/quantus/metrics/inverse_estimation.py
@@ -235,6 +235,11 @@ def __call__(
             a_batch is not None
         ), "'a_batch' must be provided to run the inverse estimation."
 
+        assert self.metric_init.abs == True, (
+            "To run the inverse estimation, you cannot set 'a_batch' to "
+            "have positive attributions only. Set 'abs' param of the metric init to 'False'."
+        )
+
         self.scores = self.metric_init(
             model=model,
             x_batch=x_batch,