修改rerank文件每次写入的数量

9fca3d4b · 张彦钊 · 3bfe07d5 · 9fca3d4b
Commit 9fca3d4b authored Sep 05, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

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

No files found.
--- a/eda/esmm/Model_pipline/rerank_esmm.py
+++ b/eda/esmm/Model_pipline/rerank_esmm.py
@@ -170,7 +170,8 @@ def to_data_base(df):


 if __name__ == "__main__":
-    users_list = get_esmm_users()
+    users_list = get_esmm_users()[:2000]
+    # TODO 删除上面的2000
    print("user number")
    print(len(users_list))