Update api.py

aimclub · Jan 7, 2024 · 2efc320 · 2efc320
1 parent 9ec2d2f
commit 2efc320
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eXNN/visualization/api.py b/eXNN/visualization/api.py
@@ -27,7 +27,7 @@ def _plot(embedding, labels):
 def reduce_dim(
     data: torch.Tensor,
     mode: str,
-    out_dim=3,
+    out_dim=2,
     neighbors=20,
 ) -> np.ndarray:
     """This function reduces data dimensionality to 2 dimensions.