correctness.py 文件源码

def main():
    "Command-line interface for running test cases."
    from argparse import ArgumentParser
    p = ArgumentParser()

    p.add_argument('--boolean', action='store_true')

    p.add_argument('--minlength', type=int, default=5)
    p.add_argument('--maxlength', type=int, default=30)
    p.add_argument('--examples', type=int, required=True)
    p.add_argument('--seed', type=int, default=None)
    p.add_argument('--grammar', choices=('medium','big'), default='medium')
    p.add_argument('--aggressive', type=float, default=0.5,
                   help='Pruning rate (zero=no pruning, one=lots of pruning).')

    args = p.parse_args()

    np.random.seed(args.seed)

    s = Setup(train=args.examples,
              grammar=args.grammar,
              maxlength=args.maxlength,
              minlength=args.minlength,
              features=False)


    test = _test_correctness_boolean if args.boolean else _test_correctness

    for i, example in enumerate(s.train):
        print colors.yellow % '=============================================================='
        print 'example: %s length: %s' % (i, example.N)
        test(example, s.grammar, args.aggressive)

    print colors.green % '=============================================================='
    print colors.green % 'DONE'
    print

    if 0:
        from arsenal.debug import ip; ip()
    else:
        pl.ioff()
        pl.show()