昇腾社区首页
中文
注册

rerank_top_k

功能描述

返回列表中得分最高的topk个对象,得分越高,越相似。

函数原型

def rerank_top_k(objs, scores)

输入参数说明

参数名

数据类型

可选/必选

说明

objs

List

必选

待排序的对象列表,列表长度取值范围[1, 1000*1000], 列表元素Document中内容page_content长度取值范围[1, 128*1024*1024]。

scores

np.ndarray

必选

待排序对象对应的得分数组,数组支持维度为1,数组长度取值范围[1, 1000*1000]。

返回值说明

数据类型

说明

List

返回k个得分最高对象。