Skip to content

Conversation

@divyegala
Copy link
Contributor

This will enable interoperability between https://github.com/rapidsai/cuml GPU HDBSCAN implementation and allow us to transfer properties from the GPU model to the CPU model, letting a user train using GPU and predict using CPU.

@lmcinnes
Copy link
Collaborator

Interoperability is definitely worthwhile. Thanks for this.

@lmcinnes lmcinnes merged commit 4cf36ec into scikit-learn-contrib:master Oct 11, 2024
@divyegala
Copy link
Contributor Author

@lmcinnes could you please cut a release with this feature commit included?

rapids-bot bot pushed a commit to rapidsai/cuml that referenced this pull request Oct 17, 2024
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 scikit-learn-contrib/hdbscan#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` Authors: - Divye Gala (https://github.com/divyegala) Approvers: - Kyle Edwards (https://github.com/KyleFromNVIDIA) - Bradley Dice (https://github.com/bdice) - Dante Gama Dessavre (https://github.com/dantegd) - William Hicks (https://github.com/wphicks) URL: #6108
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants