Sigmoid test for new tables

fastmachinelearning · Apr 25, 2022 · 7c7b7af · 7c7b7af
1 parent 0258f23
commit 7c7b7af
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/pytest/test_keras_api.py b/test/pytest/test_keras_api.py
@@ -62,7 +62,8 @@ def test_dense(backend):
 @pytest.mark.parametrize("activation_function", [Activation(activation='relu', name='Activation'),
                                                  LeakyReLU(alpha=1.0),
                                                  ELU(alpha=1.0),
-                                                 PReLU(alpha_initializer="zeros",),])
+                                                 PReLU(alpha_initializer="zeros",),
+                                                 Activation(activation='sigmoid', name='Activation')])
                                                  #ThresholdedReLU(theta=1.0)])
 @pytest.mark.parametrize('backend', ['Vivado', 'Quartus'])
 def test_activations(activation_function, backend):