Merge pull request #64 from fzi-forschungszentrum-informatik/ng_checks

Ng checks
fzi-forschungszentrum-informatik · Apr 3, 2024 · 228374d · 228374d
2 parents 850407b + 4c1fdd5
commit 228374d
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 16 deletions.
diff --git a/ClassificationModels/models/ElectricDevices/OneHotEncoder.pkl b/ClassificationModels/models/ElectricDevices/OneHotEncoder.pkl
diff --git a/TSInterpret/InterpretabilityModels/counterfactual/NativeGuideCF.py b/TSInterpret/InterpretabilityModels/counterfactual/NativeGuideCF.py
@@ -164,6 +164,7 @@ def _native_guide_wrapper(self, query, predicted_label, distance, n_neighbors):
     def _findSubarray(
         self, a, k
     ):  # used to find the maximum contigious subarray of length k in the explanation weight vector
+
         if len(a.shape) == 2:
             a = a.reshape(-1)
         n = len(a)
@@ -178,6 +179,8 @@ def _findSubarray(
             # Store the sub-array elements in the array
             for j in range(i, i + k):
                 temp.append(a[j])
+
+
 
             # Push the vector in the container
             vec.append(temp)
@@ -217,6 +220,8 @@ def _counterfactual_generator_swap(
 
         if np.any(np.isnan(most_influencial_array)):
             return np.full(individual.shape, None), None
+        if len(training_weights)==1:
+            training_weights=training_weights[0]
 
         starting_point = np.where(training_weights == most_influencial_array[0])[0][0]
 
@@ -240,6 +245,8 @@ def _counterfactual_generator_swap(
             most_influencial_array = self._findSubarray(
                 (training_weights), subarray_length
             )
+            if len(training_weights)==1:
+                training_weights=training_weights[0]
             starting_point = np.where(training_weights == most_influencial_array[0])[0][
                 0
             ]

diff --git a/TSInterpret/__version__.py b/TSInterpret/__version__.py
@@ -1,2 +1,2 @@
-VERSION = (0, 4, 4)
+VERSION = (0, 4, 5)
 __version__ = ".".join(map(str, VERSION))  # noqa: F401
diff --git a/docs/Notebooks/NunCF_torch.ipynb b/docs/Notebooks/NunCF_torch.ipynb
diff --git a/setup.py b/setup.py
@@ -40,7 +40,7 @@
     "partd==1.2.0",
     "pytz>=2021.3",
     "shap>=0.39.0,< 1.0",
-    "tensorflow>=2.9.1,< 2.15.0",
+    "tensorflow>=2.9.1,< 2.14.1",
     "keras>=2.9.0,< 3.0",
     "tsfresh>=0.18.0,< 1.0",
     "tslearn>= 0.5.2,< 1.0",