Enable distance return for NN Descent

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

680 stars 180 forks source link

Enable distance return for NN Descent #2345

Closed jinsolp closed 3 weeks ago

jinsolp commented 1 month ago

Enable NN Descent to return the distances array as well (previously only returning indices array)
- Added a return_distances flag in index_params. When set to 1 (true), allocates a distance array to return.
Test for checking distances recall compared to naive knn

copy-pr-bot[bot] commented 1 month ago

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

jinsolp commented 4 weeks ago

/ok to test

divyegala commented 4 weeks ago

/ok to test

jinsolp commented 3 weeks ago

This is no longer a breaking change

divyegala commented 3 weeks ago

/merge