search
功能描述
在数据库中检索与传入的向量相似的向量。
函数原型
def search(embeddings, k, filter_dict, **kwargs)
参数说明
参数名 |
数据类型 |
可选/必选 |
说明 |
|---|---|---|---|
embeddings |
Union[List[List[float]], List[Dict]] |
必选 |
稠密向量或稀疏向量，如果为前者则其类型为List[List[float]]，如果是后者则其类型为List[dict]。 |
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]}。 |
kwargs |
Dict |
可选 |
关键字参数，可用于指定MilvusClient search方法的关键字参数。 |
返回值说明
数据类型 |
说明 |
|---|---|
Tuple[List[List[float]], List[List[int]], List[List[List]]] |
返回3个数据，第一个数据表示相似向量的得分列表，第二个数据表示相似向量的ID列表。第3个数据表示由kwargs指定的output_fields指定的字段值， |
父主题： MilvusDB类