Commit c5406f71 authored by okiss's avatar okiss
Browse files

Update main.py

parent a59d3209
...@@ -18,37 +18,6 @@ def main(): ...@@ -18,37 +18,6 @@ def main():
trainX = np.transpose(trainX, (0, 2, 1)) trainX = np.transpose(trainX, (0, 2, 1))
logging.info(trainX.shape) logging.info(trainX.shape)
run(trainX,trainY) run(trainX,trainY)
'''
if config['model'] == 'cnn' or config['model'] == 'cnn_cluster':
logging.info("Started running CNN. If you want to run other methods please choose another model in the config.py file.")
CNN.run(trainX, trainY)
elif config['model'] == 'eegnet' or config['model'] == 'eegnet_cluster':
logging.info(
"Started running EEGNet. If you want to run other methods please choose another model in the config.py file.")
eegnet_x = np.transpose(trainX, (0, 2, 1))
logging.info(eegnet_x.shape)
eegNet.run(trainX=eegnet_x, trainY=trainY)
elif config['model'] == 'inception' or config['model'] == 'inception_cluster':
logging.info("Started running InceptionTime. If you want to run other methods please choose another model in the config.py file.")
inception.run(trainX=trainX, trainY=trainY)
elif config['model'] == 'xception' or config['model'] == 'xception_cluster':
logging.info("Started running XceptionTime. If you want to run other methods please choose another model in the config.py file.")
xception.run(trainX=trainX, trainY=trainY)
elif config['model'] == 'deepeye' or config['model'] == 'deepeye_cluster':
logging.info("Started running DeepEye. If you want to run other methods please choose another model in the config.py file.")
deepeye.run(trainX=trainX, trainY=trainY)
elif config['model'] == 'deepeye-rnn' or config['model'] == 'deepeye-rnn_cluster':
logging.info("Started running deepeye-rnn. If you want to run other methods please choose another model in the config.py file.")
deepeyeRNN.run(trainX=trainX, trainY=trainY)
else:
logging.info('Cannot start the program. Please choose one model in the config.py file')
'''
logging.info("--- Runtime: %s seconds ---" % (time.time() - start_time)) logging.info("--- Runtime: %s seconds ---" % (time.time() - start_time))
logging.info('Finished Logging') logging.info('Finished Logging')
......
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