lenoctambule/autoencoder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(autoencoder.py): __str__ method for VariationalAutoencoder class

Browse Source
This commit is contained in:
Lenoctambule
2026-04-10 15:00:04 +02:00
parent ea8a4079ac
commit 849d988de5
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/easyvae/autoencoder.py
View File

@@ -96,7 +96,7 @@ class ClassicalAutoencoder(AAutoencoder):
f"Type: {__class__.__name__}", f"Type: {__class__.__name__}",
"Encoder:", "Encoder:",
f"{self.encoder}", f"{self.encoder}",
"Decoder:" "Decoder:",
f"{self.decoder}" f"{self.decoder}"
)) ))
@@ -133,6 +133,15 @@ class VariationalAutoencoder(AAutoencoder):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.sampler = SampleLayer(self.encoder.out_size, self.lr, Identity()) self.sampler = SampleLayer(self.encoder.out_size, self.lr, Identity())
def __str__(self):
return "\n".join((
f"Type: {__class__.__name__}",
"Encoder:",
f"{self.encoder}",
"Decoder:",
f"{self.decoder}"
))
def loss(self, data_set: list[np.ndarray]) -> float: def loss(self, data_set: list[np.ndarray]) -> float:
loss = 0 loss = 0
for x in data_set: for x in data_set: