Add regularization to the auto-regression operator test using a DeepO…

…Net model
ViktorC · Jan 6, 2023 · b2a9a5c · b2a9a5c
1 parent 6f472e2
commit b2a9a5c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/operators/ml/auto_regression/test_auto_regression_operator.py b/tests/operators/ml/auto_regression/test_auto_regression_operator.py
@@ -227,7 +227,10 @@ def build_model():
             combiner_net=tf.keras.Sequential(
                 [
                     tf.keras.layers.Input(3 * diff_eq.y_dimension * 10),
-                    tf.keras.layers.Dense(diff_eq.y_dimension),
+                    tf.keras.layers.Dense(
+                        diff_eq.y_dimension,
+                        kernel_regularizer=tf.keras.regularizers.L2(l2=1e-5),
+                    ),
                 ]
             ),
         )