Merge pull request #63 from TheRealGremlin/patch-10

Update api.py
aimclub · Jan 9, 2024 · 021c2dd · 021c2dd
2 parents c6107ff + 6c14727
commit 021c2dd
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.dim() > 1 and labels.shape[labels.dim() - 1] > 1:
+            if labels.ndim > 1 and labels.shape[labels.ndim - 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))))