From 9e5cf6bd52ba112e16fe17969f731af1c7e05fd7 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 24 Jun 2019 12:47:01 -0400
Subject: [PATCH] fix
---
mhcflurry/class1_neural_network.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 80e58b88..db7be791 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -613,7 +613,7 @@ class Class1NeuralNetwork(object):
str(aa_distribution.to_dict())))
y_values = from_ic50(affinities)
- assert numpy.isnan(y_values).sum() == 0, numpy.isnan(y_values).sum()
+ assert numpy.isnan(y_values).sum() == 0, y_values
if inequalities is not None:
# Reverse inequalities because from_ic50() flips the direction
# (i.e. lower affinity results in higher y values).
@@ -642,6 +642,7 @@ class Class1NeuralNetwork(object):
if shuffle_permutation is None:
shuffle_permutation = numpy.random.permutation(len(y_values))
y_values = y_values[shuffle_permutation]
+ assert numpy.isnan(y_values).sum() == 0, y_values
peptide_encoding = peptide_encoding[shuffle_permutation]
adjusted_inequalities = adjusted_inequalities[shuffle_permutation]
for key in x_dict_without_random_negatives:
--
GitLab