From b04de69046c4f9951ee8885b688f9cb0c6caed68 Mon Sep 17 00:00:00 2001
From: Alex Rubinsteyn <alex.rubinsteyn@gmail.com>
Date: Wed, 4 May 2016 17:19:16 -0400
Subject: [PATCH] fixed hotshot network test
---
test/test_neural_nets.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/test/test_neural_nets.py b/test/test_neural_nets.py
index 0adc1cc5..e791d927 100644
--- a/test/test_neural_nets.py
+++ b/test/test_neural_nets.py
@@ -36,7 +36,9 @@ def test_make_hotshot_network_properties():
eq_(nn.loss, mse)
assert np.allclose(nn.optimizer.lr.eval(), 0.7)
print(nn.layers)
- eq_(len(nn.layers), 2 + 2 * (1 + len(layer_sizes)))
+ # since the hotshot network doesn't have an embedding layer + flatten
+ # we expect two fewer total layers than the embedding network.
+ eq_(len(nn.layers), 2 * (1 + len(layer_sizes)))
def test_make_embedding_network_small_dataset():
nn = make_embedding_network(
--
GitLab