test dist predict

44930e73 · Your Name · f0c957b5 · 44930e73
Commit 44930e73 authored Jun 19, 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
@@ -187,7 +187,7 @@ def main(te_file):
    for prob in preds:
        # ctcvr.append((prob["sample_id"][0],prob['pctcvr']))
        # str_result = str_result + str(prob["sample_id"][0]) + ":"+ prob["uid"] + ":" + prob["city"] + ":" + str(prob["cid_id"]) + ":" + str(prob['pctcvr']) + ";"
-        str_result = prob.keys()
+        str_result = list(prob.keys())
    return str_result
    # return str_result[:-1]