test

eda/esmm/Model_pipline/predict.py

@@ -180,6 +180,7 @@ def main(te_file):
     result = []
     for prob in preds:
         result.append([str(prob["uid"][0]), str(prob["city"][0]), str(prob["cid_id"][0]), str(prob['pctcvr'])])
+    return result
 
 def trans(x):
     return str(x)[2:-1] if str(x)[0] == 'b' else x

eda/esmm/Model_pipline/train.py

@@ -375,6 +375,7 @@ def main(te_files):
         result = []
         for prob in preds:
             result.append([str(prob["uid"][0]), str(prob["city"][0]), str(prob["cid_id"][0]), str(prob['pctcvr'])])
+        return result
     elif FLAGS.task_type == 'export':
         print("Not Implemented, Do It Yourself!")