Update api.py

aimclub · Jan 7, 2024 · e7a5c90 · e7a5c90
1 parent 6f7223d
commit e7a5c90
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/eXNN/visualization/api.py b/eXNN/visualization/api.py
@@ -180,6 +180,7 @@ def visualize_recurrent_layer_manifolds(
             emb_res = embedder.fit_transform(layer_output.reshape(layer_output.shape[0],
                                                                   1, layer_output.shape[1]))
             emb_res = emb_res.reshape(emb_res.shape[0], 1, -1)
+        emb_res = torch.from_numpy(emb_res)
         reducing_output = reduce_dim(data=emb_res[:, 0, :], mode=mode,
                                      out_dim=out_dim, neighbors=neighbors)
         df = pd.DataFrame(reducing_output)