infgrad

/

stella-mrl-large-zh-v3.5-1792d

like 37

我想要研究的是 我使用BM25选取3个最佳结果 然后使用咱们得模型进行重排序 但是我发现 我输入3个结果 后 得分会出现5个 我输入10个结果后 得分会得到16个 从而无法进行排序
scores = rerank_model(**inputs, return_dict=True).logits.view(-1, ).float()
#这里scores 会得到5个参数

# 获取最高分数的索引
max_score_page_idx = max_score_page_idxs[scores.cpu().numpy().argmax()]
questions[query_idx]['reference'] = 'page_' + str(max_score_page_idx + 1)