From 8fa0c2dbc1d535fcaec2cbf54e0315446aaf7a60 Mon Sep 17 00:00:00 2001 From: farizrahman4u Date: Sun, 19 Mar 2017 09:31:37 +0530 Subject: [PATCH] fix cos dense test --- tests/keras_contrib/layers/test_core.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/keras_contrib/layers/test_core.py b/tests/keras_contrib/layers/test_core.py index db9b8c0..4ef20bd 100644 --- a/tests/keras_contrib/layers/test_core.py +++ b/tests/keras_contrib/layers/test_core.py @@ -32,11 +32,11 @@ def test_cosinedense(): layer_test(core.CosineDense, kwargs={'units': 3, - 'W_regularizer': regularizers.l2(0.01), - 'b_regularizer': regularizers.l1(0.01), + 'kernel_regularizer': regularizers.l2(0.01), + 'bias_regularizer': regularizers.l1(0.01), 'regularizer': regularizers.l2(0.01), - 'W_constraint': constraints.MaxNorm(1), - 'b_constraint': constraints.MaxNorm(1)}, + 'kernel_constraint': constraints.MaxNorm(1), + 'bias_constraint': constraints.MaxNorm(1)}, input_shape=(3, 2)) X = np.random.randn(1, 20)