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

feat: re-enable loss plot

Browse Source
This commit is contained in:
Lenoctambule
2026-03-27 07:40:55 +01:00
parent 3301588f72
commit 36ffe1f3d6
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mnist_test.py
View File

@@ -14,9 +14,8 @@ def mnist_test(
x_train = np.divide(x_train, 255) x_train = np.divide(x_train, 255)
x_test = np.divide(x_train, 255) x_test = np.divide(x_train, 255)
in_len = x_train[0].flatten().shape[0] in_len = x_train[0].flatten().shape[0]
autoencoder = Autoencoder(in_len, bottleneck, 0.0001, relu) autoencoder = Autoencoder(in_len, bottleneck, 0.001, relu)
x_train = x_train[:] autoencoder.train_dataset(x_train, max_epoch, patience, display_loss=True)
autoencoder.train_dataset(x_train, max_epoch, patience)
example: np.ndarray = x_test[np.random.randint(0, len(x_test))] example: np.ndarray = x_test[np.random.randint(0, len(x_test))]
code = autoencoder.encode(example.flatten()) code = autoencoder.encode(example.flatten())
output = autoencoder.decode(code) output = autoencoder.decode(code)