Commit 067b5b8e authored by Ard Kastrati's avatar Ard Kastrati
Browse files

Resolved conflicts

parents aac28309 a7a07c94
......@@ -2,6 +2,8 @@ from config import config
from ensemble import run
import numpy as np
import scipy
from utils.utils import select_best_model, comparison_plot
from utils import IOHelper
from scipy import io
import h5py
import logging
......@@ -21,9 +23,13 @@ def main():
if config['model'] == 'eegnet' or config['model'] == 'eegnet_cluster':
trainX = np.transpose(trainX, (0, 2, 1))
logging.info(trainX.shape)
# run(trainX,trainY)
tune(trainX,trainY)
# run(trainX,trainY)
# select_best_model()
# comparison_plot(n_best = 4)
logging.info("--- Runtime: %s seconds ---" % (time.time() - start_time))
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