fix: wrong axis typo in Label's observe method

2026-04-17 02:31:33 +02:00
parent 251d66a625
commit 65c6d3bbee
1 changed files with 1 additions and 1 deletions
--- a/src/easyvae/autoencoder.py
+++ b/src/easyvae/autoencoder.py
@@ -260,7 +260,7 @@ class Label:
            self.history[self.idx] = code
            self.idx += 1
        else:
-            diffs = np.linalg.norm(self.history - code, axis=0)
+            diffs = np.linalg.norm(self.history - code, axis=1)
            idx = np.argmin(diffs)
            self.history[idx] = (self.history[idx] + code) / 2