Merge pull request #62 from TheRealGremlin/patch-9

Update api.py
aimclub · Jan 9, 2024 · c6107ff · c6107ff
2 parents 4385f0d + 7eda4a7
commit c6107ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eXNN/visualization/api.py b/eXNN/visualization/api.py
@@ -192,7 +192,7 @@ def visualize_recurrent_layer_manifolds(
         df = df.iloc[::arr_reducer, :]
         if heatmap is True:
             labels_noncat = labels
-            if labels_noncat[0].shape[0] > 1:
+            if labels.dim() > 1 and labels.shape[labels.dim() - 1] > 1:
                 labels_noncat = np.argmax(labels_noncat, axis=1)
             center = np.zeros((len(np.unique(labels_noncat)), 3))
             med_dist = np.zeros((len(np.unique(labels_noncat)), len(np.unique(labels_noncat))))