bug fix

eda/esmm/Model_pipline/dist_predict.py

@@ -161,14 +161,14 @@ def main():
 
     preds = Estimator.predict(input_fn=lambda: input_fn(te_files, 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']))
-
-    indices = []
-    for prob in preds:
-        indices.append([prob['pctr'], prob['pcvr'], prob['pctcvr']])
-    return indices
+    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:
+    #     indices.append([prob['pctr'], prob['pcvr'], prob['pctcvr']])
+    # return indices
 
 def test_map(x):
     return x * x