From d54e0cc7f3cd325645a10d5a0b668bc01e4d2847 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 1 Oct 2019 17:32:07 -0400
Subject: [PATCH] fix
---
.../data_mass_spec_benchmark/run_mhcflurry.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
index 5f0235cb..ebc06df8 100644
--- a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
+++ b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
@@ -246,14 +246,15 @@ def do_predictions(chunk_index, peptides, alleles, constant_data=None):
start = time.time()
results = {}
peptides = EncodableSequences.create(peptides)
- for allele in alleles:
+ for (i, allele) in enumerate(alleles):
+ print("Processing allele %d / %d: %0.2f sec elapsed" % (
+ i + 1, len(alleles), time.time() - start))
results[allele] = predictor.predict(
peptides=peptides,
allele=allele,
throw=False,
model_kwargs=model_kwargs).astype('float32')
- if verbose:
- print("Done predicting in", time.time() - start, "sec")
+ print("Done predicting in", time.time() - start, "sec")
return (chunk_index, results)
--
GitLab