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

Enable HDBSCAN `gpu` training and `cpu` inference #6108

Closed divyegala closed 1 month ago

divyegala commented 1 month ago

Until now, we supported all combinations of GPU/CPU interoperability except the one mentioned in the title. This was because the CPU HDBSCAN package was missing attribute setters. With https://github.com/scikit-learn-contrib/hdbscan/pull/657, attribute setters are now available which allow us to transfer GPU trained attributes to the CPU model. This feature is available as part of hdbscan=0.8.39

divyegala commented 1 month ago

/merge