昇腾社区首页
中文
注册
开发者
下载

rerank

功能描述

对混合检索结果进行加权打分并排序,返回最高得分的k个结果。

函数原型

def rerank(query, texts, batch_size)

输入参数说明

参数名

数据类型

可选/必选

说明

query

str

必选

输入的排序问题。

texts

Union[list[str], list[Document]]

必选

待排序的文本列表。

batch_size

int

可选

batch大小,默认为32,该变量暂未使用。

返回值说明

数据类型

说明

Union[list[str], list[Document]]

最高得分的k个结果。