From 2efc320b3e52be83bb87efbdccb658fc870982aa Mon Sep 17 00:00:00 2001
From: Pavel Sukachev <92699977+TheRealGremlin@users.noreply.github.com>
Date: Sun, 7 Jan 2024 20:34:49 +0400
Subject: [PATCH] Update api.py

---
 eXNN/visualization/api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/eXNN/visualization/api.py b/eXNN/visualization/api.py
index effd615..d875b30 100644
--- 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.