Commit 0992c8c8 authored by zpgeng's avatar zpgeng
Browse files

Fix small error w.r.t. main.py

parent 64a8f5dd
...@@ -16,39 +16,39 @@ def main(): ...@@ -16,39 +16,39 @@ def main():
# try: # try:
trainX, trainY = IOHelper.get_mat_data(config['data_dir'], verbose=True) trainX, trainY = IOHelper.get_mat_data(config['data_dir'], verbose=True)
if config['model'] == 'cnn': if config['model'] == 'cnn' or config['model'] == 'cnn_cluster':
logging.info("Started running CNN-1. If you want to run other methods please choose another model in the config.py file.") logging.info("Started running CNN-1. If you want to run other methods please choose another model in the config.py file.")
CNN.run(trainX, trainY) CNN.run(trainX, trainY)
elif config['model'] == 'inception': 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.") 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) inception.run(trainX=trainX, trainY=trainY)
elif config['model'] == 'eegnet': 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.") 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)) eegnet_x = np.transpose(trainX, (0, 2, 1))
logging.info(eegnet_x.shape) logging.info(eegnet_x.shape)
eegNet.run(trainX=eegnet_x, trainY=trainY) eegNet.run(trainX=eegnet_x, trainY=trainY)
elif config['model'] == 'deepeye': 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.") logging.info("Started running DeepEye. If you want to run other methods please choose another model in the config.py file.")
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'] == 'deepeye2': elif config['model'] == 'deepeye2' or config['model'] == 'deepeye2_cluster':
logging.info("Started running DeepEye2. If you want to run other methods please choose another model in the config.py file.") logging.info("Started running DeepEye2. If you want to run other methods please choose another model in the config.py file.")
deepEye2.run(trainX=trainX, trainY=trainY) deepEye2.run(trainX=trainX, trainY=trainY)
elif config['model'] == 'deepeye3': elif config['model'] == 'deepeye3' or config['model'] == 'deepeye3_cluster':
logging.info("Started running DeepEye3. If you want to run other methods please choose another model in the config.py file.") logging.info("Started running DeepEye3. If you want to run other methods please choose another model in the config.py file.")
deepEye3.run(trainX=trainX, trainY=trainY) deepEye3.run(trainX=trainX, trainY=trainY)
elif config['model'] == 'xception': 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.") 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) Xception.run(trainX=trainX, trainY=trainY)
elif config['model'] == 'deepeye-lstm': elif config['model'] == 'deepeye-lstm' or config['model'] == 'deepeye-lstm_cluster':
logging.info("Started running deepeye-lstm. If you want to run other methods please choose another model in the config.py file.") 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) RNNdeep.run(trainX=trainX, trainY=trainY)
......
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