Runnies23 commited on
Commit
d700fa2
1 Parent(s): 14aef32

Add application file

Browse files
Files changed (1) hide show
  1. inference_script.py +2 -3
inference_script.py CHANGED
@@ -35,10 +35,9 @@ def answer_question(question , model , rerankmodel , corpus_embed , corpus_list,
35
 
36
  topkindex = 15
37
  # topk15scoresimilar_BGM3 = np.argsort(ALL_final_ans_list_ALL)[:,-topkindex:]
38
- topk15scoresimilar_BGM3 = np.argsort(BGM3similarity)[:,-topkindex:]
39
 
40
-
41
- BGM3_1_retrieval = [corpus_list[i] for i in topk15scoresimilar_BGM3[0]]
42
 
43
  scores = []
44
 
 
35
 
36
  topkindex = 15
37
  # topk15scoresimilar_BGM3 = np.argsort(ALL_final_ans_list_ALL)[:,-topkindex:]
38
+ topk15scoresimilar_BGM3 = np.argsort(BGM3similarity)[-topkindex:]
39
 
40
+ BGM3_1_retrieval = [corpus_list[i] for i in topk15scoresimilar_BGM3]
 
41
 
42
  scores = []
43