Browse Source

2024-07-01
标题相似度-测试服务

罗俊辉 9 months ago
parent
commit
40cddae978
2 changed files with 4 additions and 3 deletions
  1. 3 1
      applications/match_algorithm/rank.py
  2. 1 2
      applications/schedule/search_schedule.py

+ 3 - 1
applications/match_algorithm/rank.py

@@ -37,7 +37,8 @@ def title_similarity_rank(content_title, recall_list):
     :param recall_list:
     :return:
     """
-    print(content_title)
+    print("ori_title", content_title)
+    print("unsorted title list")
     include_title_list = []
     for item in recall_list:
         video_info = item['result']
@@ -49,6 +50,7 @@ def title_similarity_rank(content_title, recall_list):
         else:
             continue
         item['title'] = title
+        print(title)
         item['score'] = jcd_title_similarity(content_title, title)
         include_title_list.append(item)
     # # include_title_list加上相似度分

+ 1 - 2
applications/schedule/search_schedule.py

@@ -324,10 +324,9 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
         data=recall_list,
         trace_id=trace_id
     )
-    # print(json.dumps(recall_list, ensure_ascii=False, indent=4))
     # 按照标题相似度排序
     ranked_list = title_similarity_rank(content_title=params['title'].split("@@")[-1], recall_list=recall_list)
-    print(params['title'].split("@@")[-1])
+    # print(params['title'].split("@@")[-1])
     for i in ranked_list:
         print(i['title'], i['score'])
     # index = 0