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

refactor: code de-dup

Browse Source
This commit is contained in:
Lenoctambule
2026-03-27 23:15:13 +01:00
parent c37d1c9c26
commit eee37c8496
4 changed files with 39 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mnist_test.py
View File

@@ -9,7 +9,7 @@ def load_mnist():
import requests
mnist_path = "./mnist.npz"
mnist_url = "https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz"
mnist_url = "https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz" # noqa
if not os.path.exists(mnist_path):
with open(mnist_path, "w+b") as f:
f.write(requests.get(mnist_url, stream=True).content)
@@ -25,7 +25,7 @@ def mnist_test(
x_train, _, x_test, _ = load_mnist()
x_train = 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].shape[0] * x_train[0].shape[0]
autoencoder = Autoencoder(in_len, bottleneck, 0.001, relu)
autoencoder.train_dataset(x_train, max_epoch, patience, display_loss=True)
example: np.ndarray = x_test[np.random.randint(0, len(x_test))]