from model.initialization import initialization

from config import conf

import argparse

def boolean_string(s):

    if s.upper() not in {'FALSE', 'TRUE'}:

        raise ValueError('Not a valid boolean string')

    return s.upper() == 'TRUE'

parser = argparse.ArgumentParser(description='Train')

parser.add_argument('--cache', default=True, type=boolean_string,

                    help='cache: if set as TRUE all the training data will be loaded at once'

                         ' before the training start. Default: TRUE')

opt = parser.parse_args()

m = initialization(conf, train=opt.cache)[0]

print("Training START")

m.fit()

print("Training COMPLETE")