Commit aa21faad authored by Your Name's avatar Your Name

test dist predict

parent c1dedec4
......@@ -186,9 +186,9 @@ def main(te_file):
str_result = ""
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 ctcvr
return str_result[:-1]
# str_result = str_result + str(prob["sample_id"][0]) + ":"+ prob["uid"] + ":" + prob["city"] + ":" + str(prob["cid_id"]) + ":" + str(prob['pctcvr']) + ";"
return prob.keys()
# return str_result[:-1]
# indices = []
# for prob in preds:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment