测试获取用户画像方法

1ed4a89f · 张彦钊 · 25e25353 · 1ed4a89f · 1ed4a89f
Commit 1ed4a89f authored 5 years ago by 张彦钊
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

rerank_esmm.py rerank_esmm.py +0 -0

No files found.
--- a/eda/esmm/Model_pipline/rerank_esmm.py
+++ b/eda/esmm/Model_pipline/rerank_esmm.py
@@ -52,7 +52,7 @@ def get_user_profile(device_id,top_k = 5):
            tags = []
            if len(tag_sort) > top_k:
                for i in range(top_k):
-                    tags.append(i[0])
+                    tags.append(tag_sort[i][0])
            else:
                for i in tag_sort:
                    tags.append(i[0])

--- a/rerank_esmm.py
+++ b/rerank_esmm.py