search
功能描述
在数据库中检索与传入的向量相似的文档。
函数原型
def search(embeddings, k, filter_dict)
参数说明
参数名 |
数据类型 |
可选/必选 |
说明 |
---|---|---|---|
embeddings |
List[List[float]] |
必选 |
向量嵌入对象。向量的数量取值范围[1, 1024 * 1024)。 |
k |
int |
可选 |
返回的相似向量的个数,取值大于0,默认为“3”,取值范围为(0, 10000]。 |
filter_dict |
Dict |
可选 |
预留,MindFAISS当前暂不支持过滤检索。 |
返回值说明
数据类型 |
说明 |
---|---|
Tuple[List[List[float]], List[List[int]]] |
返回2个数据,第一个数据表示相似向量的得分列表,第二个数据表示相似向量的ID列表。 |
父主题: MindFAISS类