rerank
功能描述
抽象函数,子类需实现。方法实现对texts中的列表与query相似性排序,最相似的排序在最前面。
函数原型
@abstractmethod def rerank(query, texts, batch_size)
输入参数说明
参数名 |
数据类型 |
可选/必选 |
说明 |
---|---|---|---|
query |
str |
必选 |
输入的排序问题。 |
texts |
List[str] |
必选 |
待排序的文本列表。 |
batch_size |
int |
可选 |
batch大小,默认为32 |
返回值说明
数据类型 |
说明 |
---|---|
np.array |
与texts对应的得分数组。 |
父主题: Reranker