Spaces:
Sleeping
Sleeping
Runnies23
commited on
Commit
•
d700fa2
1
Parent(s):
14aef32
Add application file
Browse files- 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)[
|
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 |
|