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():
# try:
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.")
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.")
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.")
eegnet_x = np.transpose(trainX, (0, 2, 1))
logging.info(eegnet_x.shape)
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.")
deepeye_x = np.transpose(trainX, (0, 2, 1))
logging.info(deepeye_x.shape)
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.")
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.")
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.")
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.")
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