change tets

rerank_esmm.py

@@ -26,13 +26,17 @@ 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)
-    if len(tags) > 5:
-        tag_score = tags[:5]
-        print(tag_score)
+    tag_sort = sorted(tags.items(), key=lambda x: x[1], reverse=True)
+    tags = []
+    if len(tag_sort) > 5:
+        for i in range(5):
+            tags.append(tag_sort[i][0])
+        print(tags)
         return tag_score
     else:
-        print(tags)
+        for i in tag_sort:
+            tags.append(tag_sort[i][0])
+            print(tags)
         return tags