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

[FEA] K-Medoids (PAM) implementation for clustering #6133

Open raulcarlomagno opened 2 weeks ago

raulcarlomagno commented 2 weeks ago

As an alternative method for kmeans clustering, the idea of having kmedoids method would be great

The BanditPAM algorithm based on Multi-Armed Bandits is a good candidate, this official C++ implementation with python bindings looks great. https://github.com/motiwari/BanditPAM

migrate this to cuda could be worth it