rerank

query

str

必选

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

texts

List[str]

必选

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

batch_size

int

可选

组batch的大小，每次会组合batch_size的texts进行embed操作，取值范围[1, 1024]，默认值为“32”。

numpy.array

和texts长度一致的numpy数组，保存每个text和query的相关性得分。