test dist predict

f0c957b5 · Your Name · aa21faad · f0c957b5
Commit f0c957b5 authored Jun 19, 2019 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

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

No files found.
--- a/eda/esmm/Model_pipline/dist_predict.py
+++ b/eda/esmm/Model_pipline/dist_predict.py
@@ -187,7 +187,8 @@ 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']) + ";"
-    return prob.keys()
+        str_result = prob.keys()
+    return str_result
    # return str_result[:-1]

    # indices = []