Merge pull request #58 from TheRealGremlin/patch-8

Update api.py
aimclub · Jan 9, 2024 · 4385f0d · 4385f0d
2 parents 94f44d5 + 83d189c
commit 4385f0d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/eXNN/visualization/api.py b/eXNN/visualization/api.py
@@ -192,6 +192,8 @@ def visualize_recurrent_layer_manifolds(
         df = df.iloc[::arr_reducer, :]
         if heatmap is True:
             labels_noncat = labels
+            if labels_noncat[0].shape[0] > 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))))
             for i in range(len(np.unique(labels_noncat))):
@@ -206,7 +208,7 @@ def visualize_recurrent_layer_manifolds(
             autosize=False,
             width=1000,
             height=1000)
-        emb_out.show(renderer="colab")
+        emb_out.close()
         emb_viz[layer] = emb_out
     return emb_viz