From eec0a1576444b95f2693e013e2ccf288d7bab4fc Mon Sep 17 00:00:00 2001
From: Alex Rubinsteyn <alex.rubinsteyn@gmail.com>
Date: Wed, 4 May 2016 16:01:23 -0400
Subject: [PATCH] small changes
---
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 85f3c744..791bcd21 100644
--- a/test/test_neural_nets.py
+++ b/test/test_neural_nets.py
@@ -8,10 +8,10 @@ import numpy as np
def test_make_embedding_network():
nn = make_embedding_network(
peptide_length=3,
+ n_amino_acids=3,
layer_sizes=[3],
activation="tanh",
loss="mse",
- embedding_input_dim=3,
embedding_output_dim=20,
optimizer=RMSprop(learning_rate=0.05))
@@ -45,10 +45,10 @@ def test_make_embedding_network():
def test_make_hotshot_network():
nn = make_hotshot_network(
peptide_length=3,
+ n_amino_acids=2,
activation="relu",
loss="mse",
layer_sizes=[4],
- n_amino_acids=2,
optimizer=RMSprop(learning_rate=0.05))
X_binary = np.array([
[True, False, True, False, True, False],
--
GitLab