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个结果。 |
父主题: MixRetrieveReranker