在 Python 中使用 ANN 索引#

构建索引#

from cuvs.neighbors import cagra

dataset = load_data()
index_params = cagra.IndexParams()

index = cagra.build(build_params, dataset)

搜索索引#

from cuvs.neighbors import cagra

queries = load_queries()

search_params = cagra.SearchParams()

index = // ... build index ...

neighbors, distances = cagra.search(search_params, index, queries, k)