Convolving in the correct axis.

pes_to_spec/model.py

@@ -444,9 +444,9 @@ class SelectRelevantLowResolution(TransformerMixin, BaseEstimator):
         #widths = np.arange(10, 50, step=5)
         #peak_idx = find_peaks_cwt(sum_low_res, widths)
         gaussian = np.exp(-0.5*(axis - sum_low_res.shape[0]//2)**2/20**2)
-        gaussian /= np.sum(gaussian, axis=1, keepdims=True)
+        gaussian /= np.sum(gaussian, axis=0, keepdims=True)
         # apply it to the data
-        smoothened = fftconvolve(sum_low_res, gaussian, mode="same", axes=1)
+        smoothened = fftconvolve(sum_low_res, gaussian, mode="same", axes=0)
         peak_idx = [np.argmax(smoothened)]
         if len(peak_idx) < 1:
             raise PromptNotFoundError()