ahmet / total-perspective-vortex

from mne.datasets import eegbci
from mne.io import read_raw_edf, concatenate_raws
import mne
import matplotlib.pyplot as plt

def main():
    files = eegbci.load_data(subjects=1, runs=[6, 10, 14], path="/sgoinfre/asezgin/mne_data/")
    raw = concatenate_raws([read_raw_edf(f, preload=True) for f in files])
    eegbci.standardize(raw)
    raw.set_montage(mne.channels.make_standard_montage("colin27_1005"))

    raw_filt = raw.copy().filter(7., 30., fir_design="firwin")
    
    raw.compute_psd(fmax=80).plot()
    raw_filt.compute_psd(fmax=80).plot()
    plt.show()


if __name__ == "__main__":
    main()