test

eda/esmm/Model_pipline/dist_predict.py

@@ -166,11 +166,15 @@ def main(te_file):
     #     for prob in preds:
     #         fo.write("%f\t%f\t%f\n" % (prob['pctr'], prob['pcvr'], prob['pctcvr']))
 
-    indices = []
+    ctcvr = []
     for prob in preds:
-        indices.append([prob['pctr'], prob['pcvr'], prob['pctcvr']])
-        print(prob['pctcvr'])
-    return indices
+        ctcvr.append(prob['pctcvr'])
+    return ctcvr
+
+    # indices = []
+    # for prob in preds:
+    #     indices.append([prob['pctr'], prob['pcvr'], prob['pctcvr']])
+    # return indices
 
 def test_map(x):
     return x * x