From 5472d6e5f8116d0f07b23f938de3312dc7e5b5f4 Mon Sep 17 00:00:00 2001 From: Jiajie Jin <55341185+ignorejjj@users.noreply.github.com> Date: Tue, 24 Dec 2024 23:38:22 +0800 Subject: [PATCH] Update active_pipeline.py --- flashrag/pipeline/active_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flashrag/pipeline/active_pipeline.py b/flashrag/pipeline/active_pipeline.py index 92e26bc..1e30225 100644 --- 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)