def parse_args(): p = DefaultHelpParser( description='Model runner for KDD 2014 kaggle competition.') p.add_argument( '--model', required=True, dest='model_name', action='store', choices=models.keys(), help='the name of the model to run' ) p.add_argument( '--submit', dest='submit', action='store_true', help='toggle on to run for a submission' ) args = p.parse_args() return args
line[1], line[2], '{:.5f}'.format(line[3]) )) if __name__ == '__main__': args = parse_args() mr = ModelRunner() model_name = args.model_name if model_name not in models: raise Exception('\n`{}` is not a valid model\n\nmodels:\n {}'.format( model_name, '\n '.join(models.keys()) )) mr.init(model_name) # train if not args.submit: predictions = mr.train_model() sname = 'output/{}-{}-{}.stats.txt'.format( datetime.utcnow().strftime('%Y%m%d%H%M%S'), model_name, git_hash ) stats = mr.stats(predictions) print('') for line in stats: print(line)