From 8141f9ee2204c28d104f9f7df229e4bf2f866e06 Mon Sep 17 00:00:00 2001 From: Michael Oliver Date: Wed, 22 Feb 2017 15:04:54 -0800 Subject: [PATCH] Update to test for scale invariance --- tests/keras_contrib/layers/test_core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/keras_contrib/layers/test_core.py b/tests/keras_contrib/layers/test_core.py index 64f4b4d..6adf010 100644 --- a/tests/keras_contrib/layers/test_core.py +++ b/tests/keras_contrib/layers/test_core.py @@ -55,7 +55,7 @@ def test_cosinedense(): model.add(core.CosineDense(1, bias=False, input_shape=(20,))) model.compile(loss='mse', optimizer='rmsprop') W = model.get_weights() - W[0] = -X.T + W[0] = -2*X.T model.set_weights(W) out = model.predict(X) assert_allclose(out, -np.ones((1, 1), dtype=K.floatx()), atol=1e-5)