Skip to content

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

test

parent 4d8012c6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 9 deletions
eda/esmm/Model_pipline/predict.py
View file @ 24ccd069
...@@ -180,13 +180,6 @@ def predict(te_file): ...@@ -180,13 +180,6 @@ def predict(te_file):
result = [] result = []
for prob in preds: for prob in preds:
result.append([str(prob["uid"][0]), str(prob["city"][0]), str(prob["cid_id"][0]), str(prob['pctcvr'])]) result.append([str(prob["uid"][0]), str(prob["city"][0]), str(prob["cid_id"][0]), str(prob['pctcvr'])])
print("result")
print(result)
print('*'*100)
df = pd.DataFrame(result, columns=["uid", "city", "cid_id", "pctcvr"])
df.head(10)
print("type result")
print(type(result))
return result return result
def trans(x): def trans(x):
...@@ -229,8 +222,8 @@ if __name__ == "__main__": ...@@ -229,8 +222,8 @@ if __name__ == "__main__":
te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/test_nearby/part-r-00000"] te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/test_nearby/part-r-00000"]
result = predict(te_files) result = predict(te_files)
# df = pd.DataFrame(result, columns=["uid", "city", "cid_id", "pctcvr"]) df = pd.DataFrame(result, columns=["uid", "city", "cid_id", "pctcvr"])
# df.head(10) print(df.head(10))
......
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