Skip to content

Commit

Permalink
Added code to run baseline and experiment func
Browse files Browse the repository at this point in the history
Added new cells that run function to extract baseline and experimental data
- Dont' forget to add new labelsequence if we added new files. Copy and paste label sequence from git project to do list
  • Loading branch information
ihgumilar committed Nov 3, 2022
1 parent b0c996d commit 2e2b86e
Showing 1 changed file with 39 additions and 0 deletions.
39 changes: 39 additions & 0 deletions EEG/pre-processing/extract_eeg.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,32 @@
# ---
# jupyter:
# jupytext:
# cell_metadata_filter: -all
# custom_cell_magics: kql
# text_representation:
# extension: .py
# format_name: percent
# format_version: '1.3'
# jupytext_version: 1.11.2
# kernelspec:
# display_name: Python 3.8.10 ('hyperscanning2_redesign_new')
# language: python
# name: python3
# ---

# %%
import mne
import pandas as pd
from tqdm import tqdm
import warnings
import os
import re

# %%
warnings.filterwarnings("ignore", category=DeprecationWarning)


# %%
def extract_baseline_eeg_data(
path_2_csv_files, path_2_save_baseline_file, labelsequence=1, bad_files=[]
):
Expand Down Expand Up @@ -267,6 +286,7 @@ def extract_baseline_eeg_data(
)


# %%
def extract_experimental_eeg_data(
path_2_csv_files,
path_2_save_experimental_file,
Expand Down Expand Up @@ -747,3 +767,22 @@ def extract_experimental_eeg_data(
print(
f"All experimental files have been saved in fif format in this path {path_2_save_experimental_file}"
)


# %% [markdown]
# ### Extract baseline data (26 files of csv)

# %%
# Extract baseline
path_raw_data = "/hpc/igum002/codes/Hyperscanning2-redesign/data/EEG/"
path_2_save_baseline = "/hpc/igum002/codes/Hyperscanning2-redesign/data/EEG/raw_baseline_data/"
extract_baseline_eeg_data(path_raw_data, path_2_save_baseline)

# %% [markdown]
# ### Extract experimental data (26 files of csv)

# %%
# Extract experimental
path_2_save_experimental_data = "/hpc/igum002/codes/Hyperscanning2-redesign/data/EEG/raw_experimental_data/"
label_sequence = [1,2,1,2,1,2,3,4,3,4,3,4,5,6,5,6,5,6,7,8,7,8,7,8,9,10]
extract_experimental_eeg_data(path_raw_data, path_2_save_experimental_data,label_sequence)

0 comments on commit 2e2b86e

Please sign in to comment.