Commit 84108d3b authored by Andrew Quinn's avatar Andrew Quinn
Browse files

Set sensible default for wavelet normalisation

parent 07052ea0
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -7,7 +7,7 @@ from scipy import signal


def morlet(x, freqs, sample_rate, win_len=4, ncycles=5, ret_basis=False,
           ret_mode='power', normalise=False):
           ret_mode='power', normalise='wikipedia'):
    """Compute a morlet wavelet time-frequency transform on a univariate dataset.

    Parameters
@@ -69,7 +69,7 @@ def morlet(x, freqs, sample_rate, win_len=4, ncycles=5, ret_basis=False,
        return cwt


def cross_morlet(x, freqs, sample_rate, win_len=4, ncycles=5, normalise=False):
def cross_morlet(x, freqs, sample_rate, win_len=4, ncycles=5, normalise='wikipedia'):
    """Compute a morlet cross wavelet time-frequency transform on a multivariate dataset.

    Parameters
@@ -102,7 +102,7 @@ def cross_morlet(x, freqs, sample_rate, win_len=4, ncycles=5, normalise=False):
    return S


def get_morlet_basis(freq, ncycles, win_len, sample_rate, normalise):
def get_morlet_basis(freq, ncycles, win_len, sample_rate, normalise='wikipedia'):
    """Compute a morlet wavelet basis set based on specified parameters.

    Parameters