change test file

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

Showing with 2 additions and 2 deletions

rerank_esmm.py rerank_esmm.py +2 -2

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