change test file

4422935a · 张彦钊 · da5fe916 · 4422935a
Commit 4422935a authored Sep 03, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

rerank_esmm.py rerank_esmm.py +1 -1

No files found.
--- a/rerank_esmm.py
+++ b/rerank_esmm.py
@@ -44,7 +44,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(tag_sort[0][0])
+                tags.append(tag_sort[i][0])
        else:
            for i in tag_sort:
                tags.append(i)