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

fix: dup load method

Browse Source
This commit is contained in:
Lenoctambule
2026-04-05 08:41:17 +02:00
parent 82d61dd10f
commit 3440de851a
2 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
autoencoder.py
View File

@@ -137,11 +137,6 @@ class VariationalAutoencoder(AAutoencoder):
loss += np.sum(np.abs(x - self.forward(x)[0])) / len(x) loss += np.sum(np.abs(x - self.forward(x)[0])) / len(x)
return loss / len(data_set) return loss / len(data_set)
def load(path: str) -> 'ClassicalAutoencoder':
path = path.removesuffix('.npy') + '.npy'
data = np.load(path, allow_pickle=True)
return data.item()
def train(self, v: np.ndarray) -> float: def train(self, v: np.ndarray) -> float:
out, _ = self.forward(v) out, _ = self.forward(v)
error = out - v error = out - v

4
layers.py
View File

@@ -59,12 +59,12 @@ class SampleLayer:
self.input = v self.input = v
self.mean = self.mean_nn.forward(v) self.mean = self.mean_nn.forward(v)
self.std = self.std_nn.forward(v) self.std = self.std_nn.forward(v)
self.eps = np.random.normal(0, 1) self.eps = np.random.normal(0, 1, self.mean.shape)
return self.eps * self.std + self.mean return self.eps * self.std + self.mean
def backprop(self, error: np.ndarray) -> np.ndarray: def backprop(self, error: np.ndarray) -> np.ndarray:
mu_error = self.mean_nn.backprop(error) mu_error = self.mean_nn.backprop(error)
std_error = self.std_nn.backprop(self.eps * error) std_error = self.std_nn.backprop(error * self.eps * self.std * 0.5)
return mu_error + std_error return mu_error + std_error