From ca306faa3f6b82005d999bc8f29ddfa5e0e4ee40 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 22 Nov 2019 08:58:09 -0500
Subject: [PATCH] fix
---
mhcflurry/class1_ligandome_predictor.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/mhcflurry/class1_ligandome_predictor.py b/mhcflurry/class1_ligandome_predictor.py
index b20b334a..df9a24d1 100644
--- a/mhcflurry/class1_ligandome_predictor.py
+++ b/mhcflurry/class1_ligandome_predictor.py
@@ -230,6 +230,10 @@ class Class1LigandomePredictor(object):
[node, auxiliary_input], name="affinities_with_auxiliary")
#layer = Dense(1, activation="linear", kernel_initializer=keras.initializers.RandomNormal(mean=0.0, stddev=1e-5), use_bias=False)
+ layer = Dense(8, activation="tanh")
+ lifted = TimeDistributed(layer, name="ligandome_hidden1")
+ node = lifted(node)
+
layer = Dense(1, activation="tanh")
lifted = TimeDistributed(layer, name="ligandome_output")
ligandome_adjustment = lifted(node)
--
GitLab