main.py

from config import general_params as params
import time
from CNN import CNN
from utils import IOHelper
from DeepEye import deepEye
from InceptionTime import inception
from EEGNet import eegNet

def main():
    start_time = time.time()
    try:
        trainX, trainY = IOHelper.get_data(verbose=True)
        testX = 0
        testY = 0
    except:
        return

    if params['model'] == 'CNN-1':
        print("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 params['model'] == 'InceptionTime':
        print("Started running InceptionTime. If you want to run other methods please choose another model in the config.py file.")
        inception.run(root_dir=params["root_dir"], trainX=trainX, trainY=trainY, testX=testX, testY=testY, verbose=True)

    elif params['model'] == 'EEGNet':
        print("Started running EEGNet. If you want to run other methods please choose another model in the config.py file.")
        eegNet.run(trainX, trainY)

    elif params['model'] == 'DeepEye':
        print("Started running DeepEye. If you want to run other methods please choose another model in the config.py file.")
        deepEye.train(trainX, trainY)

    else:
        print('Cannot start the program. Please choose one model in the config.py file')

    print("--- Runtime: %s seconds ---" % (time.time() - start_time))

if __name__=='__main__':
    main()