Commit 69fa85b0 authored by Florian RICHOUX's avatar Florian RICHOUX

Fix model


Former-commit-id: f1d659d9
parent ebcdaa38
......@@ -11,16 +11,16 @@ class LSTM64x2_Embed2_10Dense_S(AbstractModel):
super().__init__()
def get_model(self):
embed = layers.Embedding(21, 2, embeddings_initializer='glorot_uniform')
embed = layers.Embedding(21, 2, embeddings_initializer='glorot_uniform', mask_zero=True)
lstm1 = layers.LSTM(64, return_sequences=True)
lstm2 = layers.LSTM(64)
input1 = Input(shape=(None,20,), dtype=np.float32, name='protein1')
input1 = Input(shape=(None,), dtype=np.float32, name='protein1')
protein1 = embed(input1)
protein1 = lstm1(protein1)
protein1 = lstm2(protein1)
input2 = Input(shape=(None,20,), dtype=np.float32, name='protein2')
input2 = Input(shape=(None,), dtype=np.float32, name='protein2')
protein2 = embed(input2)
protein2 = lstm1(protein2)
protein2 = lstm2(protein2)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment