change tets

027ee47b · 张彦钊 · 1a474266 · 027ee47b
Commit 027ee47b authored Aug 29, 2019 by 张彦钊
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 4 deletions

rerank_esmm.py rerank_esmm.py +8 -4

No files found.
--- a/rerank_esmm.py
+++ b/rerank_esmm.py
@@ -26,12 +26,16 @@ def get_user_profile():
    tags = {}
    for i in tmp:
        tags[i["content"]] = i["score"]
-    tags = sorted(tags.items(), key=lambda x: x[1], reverse=True)
+    tag_sort = sorted(tags.items(), key=lambda x: x[1], reverse=True)
-    if len(tags) > 5:
+    tags = []
-        tag_score = tags[:5]
+    if len(tag_sort) > 5:
-        print(tag_score)
+        for i in range(5):
+            tags.append(tag_sort[i][0])
+        print(tags)
        return tag_score
    else:
+        for i in tag_sort:
+            tags.append(tag_sort[i][0])
            print(tags)
        return tags