predict add sample id

61225594 · Your Name · a84b77f8 · 61225594
Commit 61225594 authored Jun 14, 2019 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

dist_predict.py eda/esmm/Model_pipline/dist_predict.py +1 -1

No files found.
--- a/eda/esmm/Model_pipline/dist_predict.py
+++ b/eda/esmm/Model_pipline/dist_predict.py
@@ -172,7 +172,7 @@ def main(te_file):

    ctcvr = []
    for prob in preds:
-        ctcvr.append([prob["sample_id"],prob['pctcvr'][0]])
+        ctcvr.append([prob["sample_id"][0],prob['pctcvr']])
    return ctcvr

    # indices = []