Skip to content

F
ffm-baseline
Commit aa21faad authored by Your Name's avatar Your Name
Browse files
Options

test dist predict

parent c1dedec4
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
eda/esmm/Model_pipline/dist_predict.py
View file @ aa21faad
...@@ -186,9 +186,9 @@ def main(te_file): ...@@ -186,9 +186,9 @@ def main(te_file):
str_result = "" str_result = ""
for prob in preds: for prob in preds:
# ctcvr.append((prob["sample_id"][0],prob['pctcvr'])) # 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 = str_result + str(prob["sample_id"][0]) + ":"+ prob["uid"] + ":" + prob["city"] + ":" + str(prob["cid_id"]) + ":" + str(prob['pctcvr']) + ";"
# return ctcvr return prob.keys()
return str_result[:-1] # return str_result[:-1]
# indices = [] # indices = []
# for prob in preds: # 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