Skip to content
Snippets Groups Projects
Commit d54e0cc7 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 112e43a4
No related merge requests found
...@@ -246,14 +246,15 @@ def do_predictions(chunk_index, peptides, alleles, constant_data=None): ...@@ -246,14 +246,15 @@ def do_predictions(chunk_index, peptides, alleles, constant_data=None):
start = time.time() start = time.time()
results = {} results = {}
peptides = EncodableSequences.create(peptides) 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( results[allele] = predictor.predict(
peptides=peptides, peptides=peptides,
allele=allele, allele=allele,
throw=False, throw=False,
model_kwargs=model_kwargs).astype('float32') 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) return (chunk_index, results)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment