Exemplo n.º 1
0
recording_length = 30  #
pred_interval = 0.5  # when to make a prediction in seconds

fs = fs_dict[HEADSET]
classifier = CcaExtraction(fs)
pred_interval = int(fs * pred_interval)

# Load data from files
data = {}
fio = FileReader()
try:
    baseline = fio.get_data(HEADSET,
                            filter_,
                            cutoff=1,
                            notch=notch,
                            highpass=highpass,
                            bandpass=bandpass,
                            corr_freq='Baseline',
                            trial=trial,
                            session=1)
    data['Baseline'] = baseline
except:
    pass
for corr_freq in corr_frequencies:
    sample = fio.get_data(HEADSET,
                          filter_,
                          cutoff=1,
                          limit=recording_length * fs,
                          notch=notch,
                          highpass=highpass,
                          bandpass=bandpass,