BAAI

/

bge-reranker-large

like 463

Follow

Beijing Academy of Artificial Intelligence 3.8k

The score is computed via a linear layer over the last hidden state of CLS. You can use a sigmoid function to normalize the score: https://pytorch.org/docs/stable/generated/torch.sigmoid.html

+1呀，如果没有阈值的话，真的很难确定到底有没有匹配到相关的文档，不然把所有并不相关片段都扔给GPT也不合理。

Hi @mosama and @lldxhm hope the below code helps

import torch
probabilities = torch.softmax(scores, dim=0)

I'm using

import torch
probabilities = torch.tanh(torch.tensor(scores))

and check if its greater than zero