Update retrieval/retrieval.py

retrieval/retrieval.py

@@ -140,10 +140,13 @@ class Retrieval:
         scores_index = np.argsort(scores)
         
         results = []
+        ss = []
         for k in range(1, self.k+1):
             index = scores_index[-k]
-            result = {'score':scores[index], 'index':index, 'context':self.contexts[index]}
+            result = {'score_bm':scores[index], 'index':index, 'context':self.contexts[index]}
             results.append(result)
+            ss.append(scores[index])
+        print("BM25:", ss)
         return results
 
     def split(self, document):