Whitening

SpikeInterface · Nov 29, 2024 · 57a9cc9 · 57a9cc9
1 parent 0d1907f
commit 57a9cc9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/spikeinterface/sorters/internal/spyking_circus2.py b/src/spikeinterface/sorters/internal/spyking_circus2.py
@@ -163,6 +163,9 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
         whitening_kwargs["dtype"] = "float32"
         if num_channels == 1:
             whitening_kwargs["regularize"] = False
+        if whitening_kwargs["regularize"]:
+            n_jobs = job_kwargs['n_jobs']
+            whitening_kwargs["regularize_kwargs"] = {"method": "GraphicalLassoCV", "n_jobs" : n_jobs}
 
         recording_w = whiten(recording_f, **whitening_kwargs)
         noise_levels = get_noise_levels(recording_w, return_scaled=False, **job_kwargs)