wip

SpikeInterface · Oct 5, 2023 · f6e2f59 · f6e2f59
1 parent ea79325
commit f6e2f59
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/src/spikeinterface/sorters/internal/tridesclous2.py b/src/spikeinterface/sorters/internal/tridesclous2.py
@@ -20,14 +20,8 @@ class Tridesclous2Sorter(ComponentsBasedSorter):
     _default_params = {
         "apply_preprocessing": True,
         "waveforms" : {"ms_before": 0.5, "ms_after": 1.5, },
-        "filtering": {"freq_min": 300, "freq_max": 8000.0},
+        "filtering": {"freq_min": 300., "freq_max": 8000.0},
         "detection": {"peak_sign": "neg", "detect_threshold": 5, "exclude_sweep_ms": 1.5, "radius_um": 150.},
-        #~ "hdbscan_kwargs": {
-            #~ "min_cluster_size": 25,
-            #~ "allow_single_cluster": True,
-            #~ "core_dist_n_jobs": -1,
-            #~ "cluster_selection_method": "leaf",
-        #~ },
         "selection": {"n_peaks_per_channel": 5000, "min_n_peaks": 20000},
         "svd": {"n_components": 6},
         "clustering": {
@@ -184,7 +178,7 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
             original_labels,
             recording,
             features_folder,
-            method="hdbscan_on_local_pca",
+            method="local_feature_clustering",
             method_kwargs=dict(
                 # clusterer="hdbscan",
                 clusterer="isocut5",
@@ -217,7 +211,7 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
             features_folder,
             radius_um=merge_radius_um,
 
-            method="waveforms_lda",
+            method="project_distribution",
             method_kwargs=dict(
                 # neighbours_mask=neighbours_mask,
                 waveforms_sparse_mask=sparse_mask,
@@ -230,8 +224,10 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
 
                 # criteria='diptest',
                 # threshold_diptest=0.5,
-                criteria="percentile",
-                threshold_percentile=80.,
+                # criteria="percentile",
+                # threshold_percentile=80.,
+                criteria="distrib_overlap",
+                threshold_overlap=0.4,
 
                 # num_shift=0
                 num_shift=2,