change test file

rerank_esmm.py

@@ -18,16 +18,24 @@ def get_esmm_users():
     return result
 
 
-def get_user_profile():
+def get_user_profile(device_id = "9C5E7C73-380C-4623-8F48-A64C8034E315"):
     r = redis.Redis(host="172.16.40.135", port=5379, password="", db=2)
-    cl_id = "9C5E7C73-380C-4623-8F48-A64C8034E315"
-    key = "user:portrait_tags:cl_id:" + str(cl_id)
+    key = "user:portrait_tags:cl_id:" + str(device_id)
     tmp = json.loads(r.get(key).decode('utf-8'))
-    # todo 找赵晨把search word映射成tag
-    tags = {}
+    print(tmp)
+    search_tag = {}
     for i in tmp:
-        tags[i["content"]] = i["score"]
-    tag_sort = sorted(tags.items(), key=lambda x: x[1], reverse=True)
+        if i["content"] in name_tag.keys():
+            search_tag[i["content"]] = i["score"]
+            tmp.pop(i)
+    print(search_tag)
+    print("filter")
+    print(tmp)
+    tag_score = {}
+    for i in tmp:
+        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 = []
     if len(tag_sort) > 5:
         for i in range(5):
@@ -36,6 +44,8 @@ def get_user_profile():
         for i in tag_sort:
             tags.append(tag_sort[i][0])
 
+    print(tags)
+
     return tags
 
 def get_searchworlds_to_tagid():
@@ -51,7 +61,7 @@ def get_searchworlds_to_tagid():
 
         for i in tag_id:
             searchworlds_to_tagid[i[1]] = i[0]
-        print(searchworlds_to_tagid)
+
         return searchworlds_to_tagid
     except Exception as e:
         print(e)
@@ -85,6 +95,8 @@ def make_sample(uid,city_id,native_queue,nearby_queue,megacity_queue,nation_queu
 if __name__ == "__main__":
     # users_list = get_esmm_users()
     # total_samples = list()
+    name_tag = get_searchworlds_to_tagid()
+    get_user_profile()
     # for i in users_list:
     #     tag_score = get_user_profile(i[0])
     #     native,nearby,megacity,nation = get_queues(i[0],i[1])
@@ -97,7 +109,7 @@ if __name__ == "__main__":
     #
     # total_samples.todf
     # to_data_base(df)
-    get_searchworlds_to_tagid()
+