Sparsify the weights

Author: yger

src/spikeinterface/sortingcomponents/peak_detection.py

@@ -631,7 +631,7 @@ def __init__(
         weight_method={},
     ):
         PeakDetector.__init__(self, recording, return_output=True)
-
+        import scipy
         if not HAVE_NUMBA:
             raise ModuleNotFoundError('matched_filtering" needs numba which is not installed')
 
@@ -664,7 +664,7 @@ def __init__(
             self.num_templates *= 2
 
         self.weights = self.weights.reshape(self.num_templates * self.num_z_factors, -1)
-
+        self.weights = scipy.sparse.csr_matrix(self.weights)
         random_data = get_random_data_chunks(recording, return_scaled=False, **random_chunk_kwargs)
         conv_random_data = self.get_convolved_traces(random_data)
         medians = np.median(conv_random_data, axis=1)
@@ -737,7 +737,7 @@ def get_convolved_traces(self, traces):
         import scipy.signal
 
         tmp = scipy.signal.oaconvolve(self.prototype[None, :], traces.T, axes=1, mode="valid")
-        scalar_products = np.dot(self.weights, tmp)
+        scalar_products = self.weights.dot(tmp)
         return scalar_products