def main(cmd, **kwargs):
    if (cmd == 'train'):
        train(**kwargs)
    elif (cmd == 'predict'):
        predict(**kwargs)
def main(cmd, **kwargs):
    if (cmd == 'train'):
        train(**kwargs)
    elif (cmd == 'predict'):
        predict(**kwargs)