diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py index 0e038dd7321f6cb09a949735e0b56bd4fb7a7892..af76147dd829aeb057889ccd39a8d16f1a45b394 100644 --- a/pes_to_spec/model.py +++ b/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()