wip

SpikeInterface · Oct 6, 2023 · f5a42e7 · f5a42e7
1 parent ee62678
commit f5a42e7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/spikeinterface/sorters/internal/tridesclous2.py b/src/spikeinterface/sorters/internal/tridesclous2.py
@@ -29,11 +29,12 @@ class Tridesclous2Sorter(ComponentsBasedSorter):
         "waveforms": {
             "ms_before": 0.5,
             "ms_after": 1.5,
+            "radius_um": 120.0,
         },
         "filtering": {"freq_min": 300.0, "freq_max": 12000.0},
         "detection": {"peak_sign": "neg", "detect_threshold": 5, "exclude_sweep_ms": 1.5, "radius_um": 150.0},
         "selection": {"n_peaks_per_channel": 5000, "min_n_peaks": 20000},
-        "features": {"radius_um": 120},
+        "features": {},
         "svd": {"n_components": 6},
         "clustering": {
             "split_radius_um": 40.0,
@@ -155,13 +156,14 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
         #                             upsampling_um=5.0,
         #                             )
 
+        radius_um = params["waveforms"]["radius_um"]
         node3 = ExtractSparseWaveforms(
             recording,
             parents=[node0],
             return_output=True,
-            ms_beforems_before,
+            ms_before=ms_before,
             ms_after=ms_after,
-            radius_um=radius,
+            radius_um=radius_um,
         )
 
         model_folder_path = sorter_output_folder / "tsvd_model"