feat: str methods for Autoencoder

2026-03-29 09:41:33 +02:00
parent 7aabc5db48
commit 44bf4c0286
3 changed files with 10 additions and 2 deletions
--- a/mnist_test.py
+++ b/mnist_test.py
@@ -51,8 +51,7 @@ def mnist_test(filename: str):
    x_train = x_train / 255
    x_test = x_test / 255
    autoencoder: Autoencoder = Autoencoder.load(filename)
-    for i in autoencoder.encoder.layers:
-        print(len(i.input), len(i.output))
+    print(autoencoder)
    idx = np.random.randint(0, len(x_test))
    example: np.ndarray = x_test[idx]
    output, code = autoencoder.forward(example.flatten())