def parse_args(): ''' Parse cmd line arguments ''' parser = argparse.ArgumentParser() BNPYArgParser.addRequiredVizArgsToParser(parser) BNPYArgParser.addStandardVizArgsToParser(parser) parser.add_argument('--lap', default=None, type=float, help="Specific lap at which to plot parameters." \ + " If exact lap not available, instead plots nearest lap.") parser.add_argument('--doPlotData', action='store_true', default=False, help="If present, also plot training data.") parser.add_argument('--doPlotTruth', action='store_true', default=False, help="If present, also plot true model params that generated data.") parser.add_argument('--doSort', action='store_true', default=False, help="If present, sort parameters by global appearance probabilities.") args = parser.parse_args() return args
def parse_args(): ''' Returns Namespace of parsed arguments retrieved from command line ''' parser = argparse.ArgumentParser() BNPYArgParser.addRequiredVizArgsToParser(parser) BNPYArgParser.addStandardVizArgsToParser(parser) parser.add_argument('--xvar', type=str, default='laps', help="name of x axis variable to plot. one of {iters,laps,times}") parser.add_argument('--traceEvery', type=str, default=None, help="Specifies how often to plot data points. For example, traceEvery=10 only plots data points associated with laps divisible by 10.") parser.add_argument('--legendnames', type=str, default=None, help="optional names to show on legend in place of jobnames") args = parser.parse_args() args.algNames = args.algNames.split(',') args.jobnames = args.jobnames.split(',') if args.legendnames is not None: args.legendnames = args.legendnames.split(',') #assert len(args.legendnames) == len(args.jobnames) * len(args.algNames) return args