search

rapidsai / cuml

cuML - RAPIDS Machine Learning Library
https://docs.rapids.ai/api/cuml/stable/
Apache License 2.0
4.25k stars 534 forks source link

DOCSTRING - Update UMAP docstring to include approximate kNN feature #6062

Closed radinplaid closed 2 months ago

radinplaid commented 2 months ago

cuml 24.08.00 included an update to UMAP to use approximate kNN rather than exact, and the UMAP docstrings needs to be updated.

The docs suggest cuML always uses exact NN but that is only true for datasets <= 50k elements.

Files to be updated:

divyegala commented 2 months ago

@viclafargue could you please address this?

viclafargue commented 2 months ago

Thanks for opening the issue. I opened a PR to address it.