Python code

MatteoDrago · Sep 4, 2018 · 0406cf4 · 0406cf4
1 parent dc84366
commit 0406cf4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/code/main.py b/code/main.py
@@ -9,17 +9,17 @@
 from keras.callbacks import ModelCheckpoint, ReduceLROnPlateau
 from keras.utils import to_categorical
 
-# PARAMETERS #####################################################################
+# PARAMETERS ########################################################################################################
 subject = [1,2,3,4]
-task = "A"                                              # choose between "A" or "B"
-model_names = ["ConvolutionalDeepRecurrent", "Convolutional2DRecurrent"]
+task = "A"    # choose between "A" or "B"
+model_names = ["Convolutional", "Convolutional1DRecurrent", "Convolutional2DRecurrent", "ConvolutionalDeepRecurrent"]
 data_folder = "../data/full/"
 window_size = 15
 stride = 5
 GPU = True
 epochs = 10
 batch_size = 32
-##################################################################################
+#####################################################################################################################
 
 # create folder to store temporary data and results
 if not(os.path.exists("./data")):

diff --git a/code/main_multiuser.py b/code/main_multiuser.py
@@ -9,17 +9,17 @@
 from keras.callbacks import ModelCheckpoint, ReduceLROnPlateau
 from keras.utils import to_categorical
 
-# PARAMETERS #####################################################################
+# PARAMETERS ########################################################################################################
 subject = [23]
-task = "A"                                              # choose between "A" or "B"
-model_names = ["Convolutional2DRecurrent", "ConvolutionalDeepRecurrent"]
+task = "A"    # choose between "A" or "B"
+model_names = ["Convolutional", "Convolutional1DRecurrent", "Convolutional2DRecurrent", "ConvolutionalDeepRecurrent"]
 data_folder = "../data/full/"
 window_size = 15
 stride = 5
 GPU = True
 epochs = 10
 batch_size = 32
-##################################################################################
+#####################################################################################################################
 
 # create folder to store temporary data and results
 if not(os.path.exists("./data")):