From 36ffe1f3d688a75c7114e383e43a2e89be21cb21 Mon Sep 17 00:00:00 2001 From: Lenoctambule <106790775+lenoctambule@users.noreply.github.com> Date: Fri, 27 Mar 2026 07:40:55 +0100 Subject: [PATCH] feat: re-enable loss plot --- mnist_test.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mnist_test.py b/mnist_test.py index 0bd23a9..03a486b 100644 --- a/mnist_test.py +++ b/mnist_test.py @@ -14,9 +14,8 @@ def mnist_test( x_train = np.divide(x_train, 255) x_test = np.divide(x_train, 255) in_len = x_train[0].flatten().shape[0] - autoencoder = Autoencoder(in_len, bottleneck, 0.0001, relu) - x_train = x_train[:] - autoencoder.train_dataset(x_train, max_epoch, patience) + 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))] code = autoencoder.encode(example.flatten()) output = autoencoder.decode(code)