Minor in TSR

fzi-forschungszentrum-informatik · Feb 28, 2024 · 3af5311 · 3af5311
1 parent 3b7288f
commit 3af5311
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/TSInterpret/InterpretabilityModels/Saliency/SaliencyMethods_PTY.py b/TSInterpret/InterpretabilityModels/Saliency/SaliencyMethods_PTY.py
@@ -248,8 +248,8 @@ def _getTwoStepRescaling(
         inputGrad = np.zeros((input_size, 1))
         newGrad = np.zeros((input_size, sequence_length))
 
-        if self.mode == "time":
-            newGrad = np.swapaxes(newGrad, -1, -2)
+        #if self.mode == "time":
+        #    newGrad = np.swapaxes(newGrad, -1, -2)
 
 
         if hasBaseline is None:
@@ -350,8 +350,10 @@ def _getTwoStepRescaling(
         timeContribution = preprocessing.minmax_scale(timeGrad, axis=1)
 
         meanTime = np.quantile(timeContribution, 0.55)
+
         if input_size>1:
             for t in range(sequence_length):
+                print('TIME CONR',timeContribution[0, t])
                 if timeContribution[0, t] > meanTime:
                     for c in range(input_size):
                         newInput = input.clone()
@@ -410,11 +412,11 @@ def _getTwoStepRescaling(
 
                 else:
                     featureContribution = np.ones((input_size, 1)) * 0.1
+
 
-                if self.mode == "time":
-                    newGrad = np.swapaxes(newGrad, -1, -2)
                 for c in range(input_size):
                     newGrad[c, t] = timeContribution[0, t] * featureContribution[c, 0]
+
         else: 
             newGrad=timeContribution