plot_debug_sss_cov_rank_2020_03_31.py

# Author: Denis A. Engemann <denis.engemann@gmail.com>
#
# License: BSD (3-clause)import mne

epochs_fname = 'XXX'  # put real name here
raw_fname = 'YYY'  # and here.

epo_rest = mne.read_epochs(epochs_fname, proj=False, verbose=None)
raw_noise = mne.io.read_raw_fif(raw_fname, preload=True)
raw_noise.filter(2, 40, fir_design='firwin')
raw_noise.crop(0, 60)

methods = ['shrunk', 'empirical']

noise_covs = mne.compute_raw_covariance(
    raw_noise, method=methods,
    return_estimators=True)


for ii, noise_cov in enumerate(noise_covs):
    fig1, fig2 = noise_cov.plot(raw_noise.info)
    fig1.suptitle(methods[ii])
    fig2.suptitle(methods[ii])


noise_covs2 = mne.compute_raw_covariance(
    raw_noise, method=methods,
    rank=69,
    return_estimators=True)


for ii, noise_cov in enumerate(noise_covs2):
    fig1, fig2 = noise_cov.plot(raw_noise.info)
    fig1.suptitle(methods[ii])
    fig2.suptitle(methods[ii])