change test file

rerank_esmm.py

@@ -25,14 +25,14 @@ def get_user_profile(device_id = "9C5E7C73-380C-4623-8F48-A64C8034E315"):
     search_tag = {}
     for i in tmp:
         if i["content"] in name_tag.keys():
-            search_tag[i["content"]] = i["score"]
-            tmp.pop(tmp.index(i))
+            search_tag[name_tag[i["content"]]] = i["score"]
     print(search_tag)
     print("filter")
     print(tmp)
     tag_score = {}
     for i in tmp:
-        tag_score[i["content"]] = i["score"]
+        if i["type"] == "tag":
+            tag_score[i["content"]] = i["score"]
     tag_score.update(search_tag)
     tag_sort = sorted(tag_score.items(), key=lambda x: x[1], reverse=True)
     tags = []