Merge pull request #2080 from yger/sc2_fixes

Sc2 fixes
SpikeInterface · Oct 6, 2023 · 8ba35b4 · 8ba35b4
2 parents 03f0f08 + f56780d
commit 8ba35b4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/spikeinterface/sorters/internal/spyking_circus2.py b/src/spikeinterface/sorters/internal/spyking_circus2.py
@@ -114,7 +114,9 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
 
         ## We get the labels for our peaks
         mask = peak_labels > -1
-        sorting = NumpySorting.from_times_labels(selected_peaks["sample_index"][mask], peak_labels[mask], sampling_rate)
+        sorting = NumpySorting.from_times_labels(
+            selected_peaks["sample_index"][mask], peak_labels[mask].astype(int), sampling_rate
+        )
         clustering_folder = sorter_output_folder / "clustering"
         if clustering_folder.exists():
             shutil.rmtree(clustering_folder)

diff --git a/src/spikeinterface/sortingcomponents/benchmark/benchmark_clustering.py b/src/spikeinterface/sortingcomponents/benchmark/benchmark_clustering.py
@@ -524,7 +524,7 @@ def plot_statistics(self, metric="cosine", annotations=True, detect_threshold=5)
             template_real = template_real.reshape(template_real.size, 1).T
 
             if metric == "cosine":
-                dist = sklearn.metrics.pairwise.cosine_similarity(template, template_real, metric).flatten().tolist()
+                dist = sklearn.metrics.pairwise.cosine_similarity(template, template_real).flatten().tolist()
             else:
                 dist = sklearn.metrics.pairwise_distances(template, template_real, metric).flatten().tolist()
             res += dist