From 87402b975cbc5f8c95346cc591dad39feb0a4da7 Mon Sep 17 00:00:00 2001
From: Alex Rubinsteyn <alex.rubinsteyn@gmail.com>
Date: Wed, 4 May 2016 17:16:52 -0400
Subject: [PATCH] use np.allclose for learning rate comparisons
---
test/test_neural_nets.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/test/test_neural_nets.py b/test/test_neural_nets.py
index 5d4380d0..0adc1cc5 100644
--- a/test/test_neural_nets.py
+++ b/test/test_neural_nets.py
@@ -17,7 +17,7 @@ def test_make_embedding_network_properties():
optimizer=RMSprop(lr=0.7, rho=0.9, epsilon=1e-6))
eq_(nn.layers[0].input_dim, 3)
eq_(nn.loss, mse)
- eq_(nn.optimizer.lr.eval(), 0.7)
+ assert np.allclose(nn.optimizer.lr.eval(), 0.7)
print(nn.layers)
# embedding + flatten + (dense->activation) * hidden layers and last layer
eq_(len(nn.layers), 2 + 2 * (1 + len(layer_sizes)))
@@ -34,7 +34,7 @@ def test_make_hotshot_network_properties():
optimizer=RMSprop(lr=0.7, rho=0.9, epsilon=1e-6))
eq_(nn.layers[0].input_dim, 6)
eq_(nn.loss, mse)
- eq_(nn.optimizer.lr.eval(), 0.7)
+ assert np.allclose(nn.optimizer.lr.eval(), 0.7)
print(nn.layers)
eq_(len(nn.layers), 2 + 2 * (1 + len(layer_sizes)))
--
GitLab