search

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

Binarize Dice Distance for Dense Inputs #2370

Open aamijar opened 1 week ago

aamijar commented 1 week ago

Instead of binarizing in pairwise_distances() in cuml, do it in the distance function in raft.

copy-pr-bot[bot] commented 1 week 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.

cjnolet commented 1 week ago

/ok to test

cjnolet commented 3 days ago

/ok to test