Added sigmoid activation, and ability to do activation above linear layers.

2025-05-22 04:56:26 +08:00 · 2023-01-25 17:36:44 -05:00
parent 4948c0ea4e
commit 301b230c73
7 changed files with 57 additions and 10 deletions
--- a/tests/test_activation_function.py
+++ b/tests/test_activation_function.py
@ -22,7 +22,7 @@ class CombinedScene(ThreeDScene):
                ImageLayer(numpy_image, height=1.5),
                Convolutional2DLayer(1, 7, filter_spacing=0.32),
                Convolutional2DLayer(3, 5, 3, filter_spacing=0.32, activation_function="ReLU"),
-                FeedForwardLayer(3),
+                FeedForwardLayer(3, activation_function="Sigmoid"),
            ],
            layer_spacing=0.25,
        )