Update active_pipeline.py

RUC-NLPIR · Dec 24, 2024 · 5472d6e · 5472d6e
1 parent ea7b03c
commit 5472d6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flashrag/pipeline/active_pipeline.py b/flashrag/pipeline/active_pipeline.py
@@ -190,7 +190,7 @@ def judge_retrieve(self, input_prompts):
                     score_dict = {}
                     for tok, tok_id in self.ret_tokens.items():
                         if tok_id not in all_pred_log_probs[idx][0]:
-                            score_dict[tok] = -100
+                            score_dict[tok] = np.exp(-100)
                         else:
                             prob = all_pred_log_probs[idx][0][tok_id].logprob
                             score_dict[tok] = np.exp(prob)