main.py

from config import general_params as params
from IOHelper import get_data
import time
from run import run
def main():
    start_time = time.time()
    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.")
        run()
    else:
        raise Exception('Please choose one of the following models in the config.py file')

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

if __name__=='__main__':
    main()