From 214a14743c82ebd8d8245eadc77d84f6a5903252 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 2 Dec 2019 12:12:02 -0500
Subject: [PATCH] update
---
mhcflurry/class1_ligandome_predictor.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/mhcflurry/class1_ligandome_predictor.py b/mhcflurry/class1_ligandome_predictor.py
index cc851e69..e96e7124 100644
--- a/mhcflurry/class1_ligandome_predictor.py
+++ b/mhcflurry/class1_ligandome_predictor.py
@@ -625,6 +625,13 @@ class Class1LigandomePredictor(object):
fit_info["num_points"] = len(labels)
self.fit_info.append(dict(fit_info))
+ return {
+ 'batch_generator': batch_generator,
+ 'last_x': x_dict_with_random_negatives,
+ 'last_y': [encoded_y1, encoded_y2, encoded_y2],
+ 'fit_info': fit_info,
+ }
+
def predict(
self,
peptides,
--
GitLab