Merge pull request #2908 from rat-h/main

hdbscan is awfully slow on CPU side. I added hdbscan-gpu option to us…

Author: samuelgarcia

src/spikeinterface/sorters/internal/simplesorter.py

@@ -185,6 +185,11 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
 
             out = hdbscan.hdbscan(features_flat, **clust_params)
             peak_labels = out[0]
+        elif clust_method == "hdbscan-gpu":
+            from cuml.cluster import HDBSCAN as hdbscan
+
+            model = hdbscan(**clust_params).fit(features_flat)
+            peak_labels = model.labels_.copy()
         elif clust_method in ("kmeans"):
             from sklearn.cluster import MiniBatchKMeans