diff --git a/mhcflurry/class1_ligandome_predictor.py b/mhcflurry/class1_ligandome_predictor.py index b20b334ac4a47a02ff1e969aa57947cea85f60dd..df9a24d10aa6a19b371f551613a8ef02687ec838 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)