From c887a4fcd04a0b42880825cbfdf6abb834309b1a Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 19 Sep 2019 15:09:21 -0400
Subject: [PATCH] fix
---
mhcflurry/class1_ligandome_predictor.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mhcflurry/class1_ligandome_predictor.py b/mhcflurry/class1_ligandome_predictor.py
index 74a9e1db..2dc43509 100644
--- a/mhcflurry/class1_ligandome_predictor.py
+++ b/mhcflurry/class1_ligandome_predictor.py
@@ -156,8 +156,8 @@ class Class1LigandomePredictor(object):
y_true = tf.reshape(tf.cast(y_true, tf.bool), (-1,))
pos = tf.boolean_mask(y_pred, y_true)
- #pos_max = tf.reduce_max(pos, axis=1)
-
+ pos_max = tf.reduce_max(pos, axis=1)
+
neg = tf.boolean_mask(y_pred, tf.logical_not(y_true))
result = tf.reduce_sum(
tf.maximum(0.0, tf.reshape(neg, (-1, 1)) - pos_max + delta) ** 2)
--
GitLab