Captum interface and SimilarityInfluence

dilyabareeva · Jun 14, 2024 · 9767b30 · 9767b30
1 parent 0ee06fe
commit 9767b30
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 92 deletions.
diff --git a/src/explainers/aggregators/aggregators.py b/src/explainers/aggregators/aggregators.py
diff --git a/src/explainers/base.py b/src/explainers/base.py
@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Optional, Union
+from typing import List, Optional, Union
 
 import torch
 
@@ -23,7 +23,7 @@ def __init__(
         self.model.to(dev)
 
     @abstractmethod
-    def explain(self, test: torch.Tensor, **kwargs):
+    def explain(self, test: torch.Tensor, targets: Union[List[int], torch.Tensor, None], **kwargs):
         raise NotImplementedError
 
     @abstractmethod

diff --git a/tests/explainers/aggregators/test_aggregators.py b/tests/explainers/aggregators/test_aggregators.py