test

734f50a8 · Your Name · 4863e628 · 734f50a8
Commit 734f50a8 authored Jun 25, 2019 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

predict.py eda/esmm/Model_pipline/predict.py +3 -2

No files found.
--- a/eda/esmm/Model_pipline/predict.py
+++ b/eda/esmm/Model_pipline/predict.py
@@ -180,8 +180,9 @@ def predict(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'])])
-    print("result[0]")
-    print(result[0])
+    print("result")
+    print(result)
+    print('*'*100)
    df = pd.DataFrame(result, columns=["uid", "city", "cid_id", "pctcvr"])
    df.head(10)
    return result