昇腾社区首页
中文
注册

rerank

功能描述

抽象函数,子类需实现。方法实现对texts中的列表与query相似性排序,最相似的排序在最前面。

函数原型

@abstractmethod
def rerank(query, texts, batch_size)

输入参数说明

参数名

数据类型

可选/必选

说明

query

str

必选

输入的排序问题。

texts

List[str]

必选

待排序的文本列表。

batch_size

int

可选

batch大小,默认为32

返回值说明

数据类型

说明

np.array

与texts对应的得分数组。