From ad9a450eb17ba9e0913bb39bc8956c661bc08f59 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sat, 27 Jan 2018 22:42:44 -0500
Subject: [PATCH] update
---
downloads-generation/models_class1/GENERATE.sh | 2 +-
mhcflurry/class1_affinity_predictor.py | 4 ++++
2 files changed, 5 insertions(+), 1 deletion(-)
diff --git a/downloads-generation/models_class1/GENERATE.sh b/downloads-generation/models_class1/GENERATE.sh
index a63b17de..2722c9fa 100755
--- a/downloads-generation/models_class1/GENERATE.sh
+++ b/downloads-generation/models_class1/GENERATE.sh
@@ -38,7 +38,7 @@ time mhcflurry-class1-train-allele-specific-models \
--percent-rank-calibration-num-peptides-per-length 1000000 \
--min-measurements-per-allele 75 \
--train-num-jobs 0 \
- --calibration-num-jobs 4
+ --calibration-num-jobs 1
cp $SCRIPT_ABSOLUTE_PATH .
bzip2 LOG.txt
diff --git a/mhcflurry/class1_affinity_predictor.py b/mhcflurry/class1_affinity_predictor.py
index 88c807af..7f0d248f 100644
--- a/mhcflurry/class1_affinity_predictor.py
+++ b/mhcflurry/class1_affinity_predictor.py
@@ -933,6 +933,10 @@ class Class1AffinityPredictor(object):
if worker_pool and len(alleles) > 1:
# Run in parallel
+
+ # Performance hack.
+ self.neural_networks[0].peptides_to_network_input(encoded_peptides)
+
do_work = partial(
_calibrate_percentile_ranks,
predictor=self,
--
GitLab