Commit aad18f6f authored by Your Name's avatar Your Name

test

parent 99c54cdd
......@@ -157,7 +157,7 @@ def model_fn(features, labels, mode, params):
predictions=predictions,
export_outputs=export_outputs)
def main(te_file):
def predict(te_file):
dt_dir = (date.today() + timedelta(-1)).strftime('%Y%m%d')
model_dir = "hdfs://172.16.32.4:8020/strategy/esmm/model_ckpt/DeepCvrMTL/" + dt_dir
# te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/nearby/part-r-00000"]
......@@ -223,7 +223,7 @@ if __name__ == "__main__":
te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/test_nearby/part-r-00000"]
result = main(te_files)
result = predict(te_files)
df = pd.DataFrame(result, columns=["uid", "city", "cid_id", "pctcvr"])
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