def wave_hard_filter(y, sigma, tau, w): Yh =np.zeros(y.shape) if len(y.shape) == 2: for i in range(y.shape[0]): Yh[i] = utils.wave_hard_filter(y[i], sigma, tau, w) else: Yh = utils.wave_hard_filter(y, sigma, tau, w) return Yh
def filt_hard (y, tau): return utils.wave_hard_filter(y, sigma, tau, w)
for i in range(Y.shape[0]): mug = sc.opt(parameters, utils.gaussian_filter, Yb[i], Y[i]) Yg[i] = utils.gaussian_filter(Yb[i], mug*sigma) snrframe.loc[:,'gaussian'] = pd.Series(data=[utils.snr(Y, Yg), utils.snr(Y, Yg) ], index=snrframe.index) snrframe.loc[:,'noisy'] = pd.Series(data=[utils.snr(Y, Yb), utils.snr(Y, Yb) ], index=snrframe.index) for w in pywt.wavelist('db'): print w def filt_hard (y, tau): return utils.wave_hard_filter(y, sigma, tau, w) def filt_soft(y, tau): return utils.wave_soft_filter(y, sigma, tau, w) for i in range(Y.shape[0]): tauhard = sc.opt(parameters, filt_hard, Yb[i], Y[i]) tausoft = sc.opt(parameters, filt_soft, Yb[i], Y[i]) Yh[i] = utils.wave_hard_filter(Yb[i], sigma, tauhard, w) Ys[i] = utils.wave_soft_filter(Yb[i], sigma, tausoft, w) snrframe.loc[:,w] = pd.Series(data=[utils.snr(Y, Yh), utils.snr(Y, Ys) ], index=snrframe.index) for w in pywt.wavelist('bior'): print w def filt_hard (y, tau): return utils.wave_hard_filter(y, sigma, tau, w) def filt_soft(y, tau): return utils.wave_soft_filter(y, sigma, tau, w) for i in range(Y.shape[0]): tauhard = sc.opt(parameters, filt_hard, Yb[i], Y[i]) tausoft = sc.opt(parameters, filt_soft, Yb[i], Y[i]) Yh[i] = utils.wave_hard_filter(Yb[i], sigma, tauhard, w) Ys[i] = utils.wave_soft_filter(Yb[i], sigma, tausoft, w) snrframe.loc[:,w] = pd.Series(data=[utils.snr(Y, Yh), utils.snr(Y, Ys) ], index=snrframe.index)