昇腾社区首页
中文
注册

search

功能描述

在数据库中检索与传入的向量相似的向量。

函数原型

def search(embeddings, k, filter_dict)

参数说明

参数名

数据类型

可选/必选

说明

embeddings

Union[List[List[float]], List[Dict[int, float]]]

必选

稠密向量或稀疏向量,如果为前者则其类型为List[List[float]],如果是后者则其类型为List[Dict[int, float]]。

k

int

可选

返回的相似向量的个数,取值大于0,默认为“3”,取值范围(0, 10000]。

filter_dict

Dict

可选

由检索条件组成的字典,filter_dict包含元素的个数需小于等于32,当前只支持对document_id进行过滤,过滤的文档id以列表形式传入,例如,需要在document_id为1,2,4这几篇文档中过滤,则传入的字典为{"document_id": [1, 2, 4]}

返回值说明

数据类型

说明

Tuple[List[List[float]], List[List[int]]]

返回2个数据,第一个数据表示相似向量的得分列表,第二个数据表示相似向量的ID列表。