在 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)