predict add sample id

67ef1075 · Your Name · 61225594 · 67ef1075
Commit 67ef1075 authored Jun 14, 2019 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

dist_predict.py eda/esmm/Model_pipline/dist_predict.py +2 -1

No files found.
--- a/eda/esmm/Model_pipline/dist_predict.py
+++ b/eda/esmm/Model_pipline/dist_predict.py
@@ -210,7 +210,8 @@ if __name__ == "__main__":
    print(rdd_te_files.collect())
    print("-" * 100)
    indices = rdd_te_files.repartition(100).map(lambda x: main(x))
-    print(indices.take(1))
+    print(indices.take(2))
+


    print("耗时(秒)：")