config.py 728 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# configuration used by the training and evaluation methods
# let's keep it here to have a clean code on other methods that we try
general_params = {}
CNN_params = {}

"""
Models:

CNN-1: First try: CNN to predict movement towards left or right (prosaccade) with 1 second data.
RNN? Transformers, Attention, etc etc
"""

# general parameters
general_params['model'] = "CNN-1"
Ard Kastrati's avatar
Ard Kastrati committed
15
general_params['data_path'] = "/Volumes/methlab/ETH_AS/preprocessed2/"
16
17

# CNN - 1
Ard Kastrati's avatar
Ard Kastrati committed
18
19
20
21
22
23
24
CNN_params['trainX_variable1'] = "EEGprocue"
CNN_params['trainX_variable2'] = "data"
CNN_params['trainX_filename'] = "EEGprocue"

CNN_params['trainY_variable1'] = "trialinfopro"
CNN_params['trainY_variable2'] = "cues"
CNN_params['trainY_filename'] = "trialinfocuelocked"