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

rerank

功能描述

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

函数原型

def rerank(query, texts, batch_size)

输入参数说明

参数名

数据类型

可选/必选

说明

query

str

必选

问题,会和所有的text进行相关性计算。字符串长度范围[1, 1024 * 1024]。

texts

list[Document]

必选

文本列表,列表长度取值(0, 1000*1000],字符串长度范围[1, 1024 * 1024]。

batch_size

int

可选

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

返回值说明

数据类型

说明

list[Document]

最高得分的k个结果。