Commit a59d3209 authored by okiss's avatar okiss
Browse files

Update main.py

parent 6529c1ed
...@@ -12,10 +12,7 @@ def main(): ...@@ -12,10 +12,7 @@ def main():
logging.info('Started the Logging') logging.info('Started the Logging')
start_time = time.time() start_time = time.time()
# try: # try:
# trainX, trainY = IOHelper.get_mat_data(config['data_dir'], verbose=True) trainX, trainY = IOHelper.get_mat_data(config['data_dir'], verbose=True)
f = io.loadmat('trainX.mat')
trainX = f['trainX'].reshape(-1,500,129)[:20,...]
trainY=io.loadmat('trainY.mat')['trainY'][:20]
if config['model'] == 'eegnet' or config['model'] == 'eegnet_cluster': if config['model'] == 'eegnet' or config['model'] == 'eegnet_cluster':
trainX = np.transpose(trainX, (0, 2, 1)) trainX = np.transpose(trainX, (0, 2, 1))
......
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