瀏覽代碼

取消 score_2的逻辑计算, 只取score1

罗俊辉 1 年之前
父節點
當前提交
751955b580
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      applications/functions/calculate.py

+ 2 - 2
applications/functions/calculate.py

@@ -115,13 +115,13 @@ def title_mix(title_p, dt, trace_id):
                 L.append(os.path.join(json_path, file))
     print("召回的视频量", len(L))
     score_list_1 = []
-    score_list_2 = []
+    # score_list_2 = []
     for file in L:
         file_name = file.split('/')[-1].replace(".json", "")
         v_id = file_name.split('_')[1]
         uid = file_name.split('_')[0]
         # score1, score2 = compute_similarity(title_p, file)
-        score1 = compute_similarity(file, file)
+        score1 = compute_similarity(title_p, file)
         score_list_1.append([score1, v_id, uid])
         # score_list_2.append([score2, v_id, uid])