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/autoencoder.py
+++ b/autoencoder.py
@@ -17,6 +17,9 @@ class Autoencoder:
        self.encoder = DeepNNLayer(encoder_layers, lr, activation_func)
        self.decoder = DeepNNLayer(decoder_layers, lr, activation_func)

+    def __str__(self):
+        return f'Encoder:\n{self.encoder}\n\nDecoder:\n{self.decoder}'
+
    def loss(self, data_set: list[np.ndarray]) -> float:
        loss = 0
        for x in data_set: