main.py 616 Bytes
Newer Older
1
2
from config import general_params as params
import time
3
4
import CNN
import IOHelper as IO
5

6
7
def main():
    start_time = time.time()
8
    try:
9
        trainX, trainY = IO.get_data(verbose=True)
10
11
12
    except:
        return

13
14
    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.")
15
        CNN.run(trainX, trainY)
16
    else:
17
        print 'Cannot start the program. Please choose one model in the config.py file'
18
19
20
21
22

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

if __name__=='__main__':
    main()