Commit 9935e1d4 authored by Alex Fout's avatar Alex Fout

added script to generate feature files for pre_test and post_test for each patient

parent d5348842
from config import pid_noConcussion, pid_3stepProtocol, pid_testRetest, pid_concussion, feature_functions
from patient import Patient
# go through each list of ids
for lst in [pid_noConcussion, pid_3stepProtocol, pid_testRetest, pid_concussion]:
# for each id...
for pid in lst:
print("Processing pid: {}".format(pid))
p = Patient(pid)
# generate file for pre_test
if p.pre_test is not None:
p.pre_test.remove_artifacts()
p.pre_test.get_examples(feature_functions, epoch_size=512) # 2 seconds
p.pre_test.save_examples()
if p.post_test is not None:
# generate file for post_test
p.post_test.remove_artifacts()
p.post_test.get_examples(feature_functions, epoch_size=512) # 2 seconds
p.post_test.save_examples()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment