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

fix: bad type hint and typo for forward method return value

Browse Source
This commit is contained in:
Lenoctambule
2026-04-05 01:44:10 +02:00
parent 5a8fb2c48b
commit 82d61dd10f
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
autoencoder.py
View File

@@ -143,7 +143,7 @@ class VariationalAutoencoder(AAutoencoder):
return data.item()
def train(self, v: np.ndarray) -> float:
out = self.forward(v)
out, _ = self.forward(v)
error = out - v
self.encoder.backprop(
self.sampler.backprop(
@@ -152,7 +152,7 @@ class VariationalAutoencoder(AAutoencoder):
)
return np.sum(np.abs(error)) / len(v)
def forward(self, v: np.ndarray) -> np.ndarray:
def forward(self, v: np.ndarray) -> tuple[np.ndarray, np.ndarray]:
code = self.encoder.forward(v)
sample = self.sampler.forward(code)
out = self.decoder.forward(sample)