predict add sample id

eda/esmm/Model_pipline/train.py

@@ -362,7 +362,7 @@ def main(_):
         preds = Estimator.predict(input_fn=lambda: input_fn(te_files, num_epochs=1, batch_size=FLAGS.batch_size), predict_keys=["pctcvr","pctr","pcvr","sample_id"])
         with open(FLAGS.local_dir + "/pred.txt", "w") as fo:
             for prob in preds:
-                fo.write("%f\t%f\t%f\t%d\n" % (prob['pctr'], prob['pcvr'], prob['pctcvr'], prob["sample_id"]))
+                fo.write("%f\t%f\t%f\t%s\n" % (prob['pctr'], prob['pcvr'], prob['pctcvr'], prob["sample_id"]))
     elif FLAGS.task_type == 'export':
         print("Not Implemented, Do It Yourself!")