rapidsai / raft

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.
https://docs.rapids.ai/api/raft/stable/
Apache License 2.0
683 stars 180 forks source link

[FEA] NN Descent + CAGRA should support additional distance metrics #2276

Open tarang-jain opened 2 months ago

tarang-jain commented 2 months ago

CAGRA graph construction with IVF-PQ will support inner product with PR https://github.com/rapidsai/raft/pull/2260. The same needs to be done for NN Descent. This is not done currently because NN Descent does not support graph building with metrics other than L2Expanded.