Commit aa01b2e1 authored by zpgeng's avatar zpgeng
Browse files

Add RNNDeep.py

parent 233a479b
......@@ -53,6 +53,8 @@ config['inception'] = {}
config['deepeye'] = {}
# EEGNet
config['eegnet'] = {}
# LSTM-DeepEye
config['deepeye-lstm'] = {}
config['inception']['input_shape'] = (125, 129) if config['downsampled'] else (500, 129)
config['eegnet']['channels'] = 129
......@@ -60,6 +62,7 @@ config['eegnet']['samples'] = 125 if config['downsampled'] else 500
config['deepeye']['input_shape'] = (129, 125) if config['downsampled'] else (129, 500)
config['deepeye']['channels'] = 129
config['deepeye']['samples'] = 125 if config['downsampled'] else 500
config['deepeye-lstm']['input_shape'] = (125, 129) if config['downsampled'] else (500, 129)
# You can set descriptive experiment names or simply set empty string ''.
# config['model_name'] = 'skeleton_code'
......
......@@ -40,6 +40,10 @@ def main():
deepeye_x = np.transpose(trainX, (0, 2, 1))
logging.info(deepeye_x.shape)
deepEye.run(trainX=deepeye_x, trainY=trainY)
elif config['model'] == 'deepeye-lstm':
logging.info("Started running deepeye-lstm. If you want to run other methods please choose another model in the config.py file.")
RNNdeep.run(trainX=trainX, trainY=trainY)
else:
logging.info('Cannot start the program. Please choose one model in the config.py file')
......
Supports Markdown
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