main.py 529 Bytes
Newer Older
1
2
3
from config import general_params as params
from IOHelper import get_data
import time
Ard Kastrati's avatar
Ard Kastrati committed
4
from run import run
5
6
7
8
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.")
Ard Kastrati's avatar
Ard Kastrati committed
9
        run()
10
11
12
13
14
15
16
    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()