feat: ActivationFunc classes ReLU and LeakyReLU

2026-03-29 19:20:05 +02:00
parent 44bf4c0286
commit 53c7f73055
5 changed files with 37 additions and 21 deletions
--- a/mnist_test.py
+++ b/mnist_test.py
@@ -1,7 +1,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 from autoencoder import Autoencoder
-from utils import leaky_relu
+from activations import LeakyReLU


 def load_mnist() -> list[np.ndarray]:
@@ -32,7 +32,7 @@ def mnist_train(
        [in_len, 64, 16],
        [16, 64, in_len],
        0.01,
-        leaky_relu
+        LeakyReLU()
    )
    autoencoder.train_dataset(
        x_train,