Skip to content

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

bug fix

parent a67b6780
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
eda/esmm/Model_pipline/dist_predict.py
View file @ c319f8c7
......@@ -161,9 +161,9 @@ def main(te_file):
preds = Estimator.predict(input_fn=lambda: input_fn(te_file, num_epochs=1, batch_size=10000), predict_keys=["pctcvr","pctr","pcvr"])
with open("/home/gmuser/esmm/nearby/pred.txt", "w") as fo:
for prob in preds:
fo.write("%f\t%f\t%f\n" % (prob['pctr'], prob['pcvr'], prob['pctcvr']))
# with open("/home/gmuser/esmm/nearby/pred.txt", "w") as fo:
# for prob in preds:
# fo.write("%f\t%f\t%f\n" % (prob['pctr'], prob['pcvr'], prob['pctcvr']))
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