Skip to content

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

test dist predict

parent 44930e73
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
eda/esmm/Model_pipline/dist_predict.py
View file @ 59a79433
......@@ -176,7 +176,7 @@ def main(te_file):
log_step_count_steps=100, save_summary_steps=100)
Estimator = tf.estimator.Estimator(model_fn=model_fn, model_dir=model_dir, params=model_params, config=config)
preds = Estimator.predict(input_fn=lambda: input_fn(te_file, num_epochs=1, batch_size=10000), predict_keys=["pctcvr","pctr","pcvr","sample_id"])
preds = Estimator.predict(input_fn=lambda: input_fn(te_file, num_epochs=1, batch_size=10000), predict_keys=["pctcvr","pctr","pcvr","sample_id","uid","city","cid_id"])
# with open("/home/gmuser/esmm/nearby/pred.txt", "w") as fo:
# 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