From 4a9b397a42404b514e82a69b8ef32ff25ccdb02b Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 27 Jan 2020 13:40:21 -0500
Subject: [PATCH] fix
---
downloads-generation/data_evaluation/join_with_precomputed.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/downloads-generation/data_evaluation/join_with_precomputed.py b/downloads-generation/data_evaluation/join_with_precomputed.py
index bcd1b986..6164fbd6 100644
--- a/downloads-generation/data_evaluation/join_with_precomputed.py
+++ b/downloads-generation/data_evaluation/join_with_precomputed.py
@@ -118,7 +118,8 @@ def run():
prediction_df.index, allele
]
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
+ df.loc[sub_df.index, name + "_best_allele"] = prediction_df.idxmax(
+ 1, skipna=False).values
if 'netmhcpan4.ba' in args.predictors:
# unflip the values
--
GitLab