From 708d63a7ec4eb8e4681ef6ce33a08e7120df9689 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 27 Jan 2020 11:04:39 -0500
Subject: [PATCH] fix
---
.../data_evaluation/join_with_precomputed.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/downloads-generation/data_evaluation/join_with_precomputed.py b/downloads-generation/data_evaluation/join_with_precomputed.py
index 6125490f..bcd1b986 100644
--- a/downloads-generation/data_evaluation/join_with_precomputed.py
+++ b/downloads-generation/data_evaluation/join_with_precomputed.py
@@ -83,6 +83,7 @@ def run():
index=peptides,
columns=["%s affinity" % a for a in alleles])).rename(
columns=lambda s: s.replace("affinity", "").strip())
+ precomputed_dfs['netmhcpan4.ba'] *= -1 # flip since it's affinities
if 'netmhcpan4.el' in args.predictors:
precomputed_dfs['netmhcpan4.el'] = load_results(
@@ -119,6 +120,10 @@ def run():
df.loc[sub_df.index, name] = prediction_df.max(1, skipna=False).values
df.loc[sub_df.index, name + " allele"] = prediction_df.idxmax(1, skipna=False).values
+ if 'netmhcpan4.ba' in args.predictors:
+ # unflip the values
+ df['netmhcpan4.ba'] *= -1
+
print("Skip experiments", skip_experiments)
print("results")
print(df)
--
GitLab