wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-09-09 11:25:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

change atol on allclose

Browse Source
This commit is contained in:
Michael Oliver
2017-02-22 11:59:38 -08:00
parent 000927f7c9
commit 538b9cd45a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/layers/test_core.py
+2 -2
View File
@@ -48,7 +48,7 @@ def test_cosinedense():
W[1] = np.asarray([1.])
model.set_weights(W)
out = model.predict(X)
assert_allclose(out, np.ones((1, 1), dtype=K.floatx()))
assert_allclose(out, np.ones((1, 1), dtype=K.floatx()), atol=1e-5)
X = np.random.randn(1, 20)
model = Sequential()
@@ -58,7 +58,7 @@ def test_cosinedense():
W[0] = -X.T
model.set_weights(W)
out = model.predict(X)
assert_allclose(out, -np.ones((1, 1), dtype=K.floatx()))
assert_allclose(out, -np.ones((1, 1), dtype=K.floatx()), atol=1e-5)
if __name__ == '__main__':