Julius/semesterproject_lecture_eeg
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Fixed some smaller things

Browse Source
This commit is contained in:
Julius Voggesberger
2021-03-28 04:46:37 +02:00
parent 3b837869fe
commit c86d46f447
13 changed files with 80 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
preprocessing_and_cleaning.py
View File

@@ -70,8 +70,7 @@ def clean_data(raw, subject, dataset, cleaned=False):
:return: the bad channels
"""
channels = None
folder = "Dataset\\" + dataset + "\\sub-" + subject + "\\ses-" + dataset + "\\eeg\\"
filepath = folder + "sub-" + subject + "_task-" + dataset
filepath = "Dataset/preprocessed/sub-" + subject + "_task-" + dataset + "_badannotations.csv"
# If nothing was marked yet, plot the data to mark bad segments
if not cleaned:
@@ -80,15 +79,15 @@ def clean_data(raw, subject, dataset, cleaned=False):
bad_idx = [idx for (idx, annot) in enumerate(raw.annotations) if annot['description'] == "BAD_"]
# If bad intervals were found save
if bad_idx:
raw.annotations[bad_idx].save(filepath + "_badannotations.csv")
raw.annotations[bad_idx].save(filepath)
if os.path.isfile(filepath + "_badannotations.csv"):
annotations = mne.read_annotations(filepath + "_badannotations.csv")
if os.path.isfile(filepath):
annotations = mne.read_annotations(filepath)
raw.annotations.append(annotations.onset, annotations.duration, annotations.description)
# Set the bad channels for each subject
if subject == '001':
channels = ['F8'] # Maybe also FP2?
channels = ['F8']
elif subject == '003':
channels = []
elif subject == '014':