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)