diff --git a/neurodsp/timefrequency/wavelets.py b/neurodsp/timefrequency/wavelets.py index 3cd76749..6fcb4c82 100644 --- a/neurodsp/timefrequency/wavelets.py +++ b/neurodsp/timefrequency/wavelets.py @@ -11,7 +11,7 @@ ################################################################################################### @multidim() -def compute_wavelet_transform(sig, fs, freqs, n_cycles=7, scaling=0.5, norm='sss'): +def compute_wavelet_transform(sig, fs, freqs, n_cycles=7, scaling=0.5, norm='amp'): """Compute the time-frequency representation of a signal using morlet wavelets. Parameters @@ -40,6 +40,12 @@ def compute_wavelet_transform(sig, fs, freqs, n_cycles=7, scaling=0.5, norm='sss mwt : 2d array Time frequency representation of the input signal. + Notes + ----- + + * This computes the continuous wavelet transform at the specified frequencies and + along all shifts. + Examples -------- Compute a Morlet wavelet time-frequency representation of a signal: