KMeans_CUDA with custom metric mahalanobis distance

src-d / kmcuda

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA

Other

783 stars 144 forks source link

KMeans_CUDA with custom metric mahalanobis distance #96

Open chaitrasj opened 4 years ago

chaitrasj commented 4 years ago

Hello, I wanted to use k means gpu code for clustering. However, I have a special requirement to use a custom distance metric i.e mahalanobis distance. Can you please guide me what changes need to be done to introduce custom metric in existing code?

Thank you.