analysis.py 文件源码

def main():
    from argparse import ArgumentParser
    p = ArgumentParser()
    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('--profile', action='store_true')
    p.add_argument('--grammar', choices=('medium','big'), default='medium')
    p.add_argument('--aggressive', type=float, default=0,
                   help='Pruning rate (zero=no pruning, one=lots of pruning).')

    args = p.parse_args()

    if args.profile:
        profile_run(examples = args.examples,
                    grammar = args.grammar,
                    maxlength = args.maxlength,
                    minlength = args.minlength,
                    aggressive = args.aggressive,
                    seed = args.seed)

    else:
        d = run(examples = args.examples,
                grammar = args.grammar,
                maxlength = args.maxlength,
                minlength = args.minlength,
                aggressive = args.aggressive,
                seed = args.seed)

        filename_base = 'tmp/cp-analysis-' + '-'.join('%s_%s' % (k,v) for k,v in sorted(args.__dict__.items()))
        d.to_csv('%s.csv' % filename_base)
        p = sns.jointplot('ratio', 'speedup', d, kind='reg')
        p.savefig('%s.png' % filename_base)
        print '[info] wrote %s.csv' % filename_base

    print '== DONE =='

    pl.ioff()
    pl.show()