# l_trans_bandwidth=1, h_trans_bandwidth=1,
# fir_design='firwin'),
# filt_params_noise_stop=dict(l_freq=freqs_noise2[1], h_freq=freqs_noise2[0],
# l_trans_bandwidth=1, h_trans_bandwidth=1,
# fir_design='firwin'),
# sampling_freq=sf, picks=picks, rank="full")
ssd = SSD(filt_params_signal=dict(l_freq=freqs_sig[0], h_freq=freqs_sig[1],
l_trans_bandwidth=1, h_trans_bandwidth=1,
fir_design='firwin'),\
filt_params_noise=dict(l_freq=freqs_noise[0], h_freq=freqs_noise[1],
l_trans_bandwidth=1, h_trans_bandwidth=1,
fir_design='firwin'),
sampling_freq=sf, picks=picks, rank="full", n_fft=4096)
#%%
ssd.fit(raw.copy().crop(0, 120))
#%%
ssd_sources = ssd.transform(raw)
#%%
psd, freqs = mne.time_frequency.psd_array_welch(ssd_sources,
sfreq=raw.info['sfreq'],
n_fft=4096)
# psd, freqs = mne.time_frequency.psd_array_welch(
# raw.get_data(), sfreq=raw.info['sfreq'], n_fft=int(np.ceil(raw.info['sfreq']/2)))
#%%
spec_ratio = ssd.spec_ratio
sorter = ssd.sorter_spec
# plot spectral ratio (see Eq. 24 in Nikulin 2011)
