diff --git a/test_bispectrum.py b/test_bispectrum.py index f580019f..fd680ec7 100644 --- a/test_bispectrum.py +++ b/test_bispectrum.py @@ -30,6 +30,7 @@ ) import mne +import numpy as np # %% # Let's read the example using `mne_bids `_. @@ -39,8 +40,14 @@ RUN_NAME = "TestBispectrum" raw = mne.io.read_raw_brainvision(r"C:\Users\ICN_admin\Documents\Datasets\Berlin\sub-002\ses-EcogLfpMedOff01\ieeg\sub-002_ses-EcogLfpMedOff01_task-SelfpacedRotationR_acq-StimOff_run-01_ieeg.vhdr") +raw = mne.io.read_raw_brainvision(r"C:\Users\ICN_admin\Documents\Datasets\Berlin\sub-003\ses-EcogLfpMedOff01\ieeg\sub-003_ses-EcogLfpMedOff01_task-SelfpacedRotationR_acq-StimOff_run-01_ieeg.vhdr") +raw = mne.io.read_raw_brainvision(r"C:\Users\ICN_admin\Documents\Datasets\Berlin\sub-005\ses-EcogLfpMedOff01\ieeg\sub-005_ses-EcogLfpMedOff01_task-SelfpacedRotationR_acq-StimOff_run-01_ieeg.vhdr") +raw = mne.io.read_raw_brainvision(r"C:\Users\ICN_admin\Documents\Datasets\Berlin\sub-005\ses-EcogLfpMedOff02\ieeg\sub-005_ses-EcogLfpMedOff02_task-SelfpacedRotationR_acq-StimOn_run-01_ieeg.vhdr") -raw.pick(["ECOG_L_1_SMC_AT", "SQUARED_ROTATION"]) + +#raw.pick(["ECOG_L_1_SMC_AT", "SQUARED_ROTATION"]) +raw.pick(["ECOG_L_1_SMC_AT"]) +raw.pick(["ECOG_R_1_SMC_AT"]) sfreq = raw.info["sfreq"] data = raw.get_data()#[:, :int(sfreq)*100] @@ -112,6 +119,17 @@ folder_name=RUN_NAME, ) +features_plt = features[np.logical_and(features["time"] > int(sfreq)*10, features["time"] < int(sfreq)*15)] +plt.figure() +plt.subplot(211) +plt.plot(data[0, int(sfreq)*10 : int(sfreq)*15]) +plt.subplot(212) +plt.plot(features_plt["ECOG_L_1_SMC_AT_Bispectrum_phase_mean_whole_fband_range"], label="phase") +plt.plot(features_plt["ECOG_L_1_SMC_AT_Bispectrum_real_mean_whole_fband_range"], label="real") +plt.plot(features_plt["ECOG_L_1_SMC_AT_Bispectrum_imag_mean_whole_fband_range"], label="imag") +plt.legend() +plt.show() + # %% # Feature Analysis Movement # -------------------------