Commit aa01b2e1 authored by zpgeng's avatar zpgeng
Browse files

Add RNNDeep.py

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