feat(autoencoder.py): __str__ method for VariationalAutoencoder class
This commit is contained in:
@@ -96,7 +96,7 @@ class ClassicalAutoencoder(AAutoencoder):
|
||||
f"Type: {__class__.__name__}",
|
||||
"Encoder:",
|
||||
f"{self.encoder}",
|
||||
"Decoder:"
|
||||
"Decoder:",
|
||||
f"{self.decoder}"
|
||||
))
|
||||
|
||||
@@ -133,6 +133,15 @@ class VariationalAutoencoder(AAutoencoder):
|
||||
super().__init__(*args, **kwargs)
|
||||
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:
|
||||
loss = 0
|
||||
for x in data_set:
|
||||
|
||||
Reference in New Issue
Block a user