From d3f5931ab158f47d1a599bd7249f37d7cecd809c Mon Sep 17 00:00:00 2001 From: shivesh Date: Fri, 11 May 2018 16:54:32 -0500 Subject: Fri May 11 16:54:32 CDT 2018 --- plot_sens.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) (limited to 'plot_sens.py') diff --git a/plot_sens.py b/plot_sens.py index 279d66a..8357ffc 100755 --- a/plot_sens.py +++ b/plot_sens.py @@ -144,6 +144,10 @@ def parse_args(args=None): '--plot', type=misc_utils.parse_bool, default='True', help='Make sensitivity plots' ) + parser.add_argument( + '--data', type=misc_utils.parse_bool, default='False', + help='Use HESE7 data' + ) parser.add_argument( '--plot-statistic', type=misc_utils.parse_bool, default='False', help='Plot MultiNest evidence or LLH value' @@ -227,6 +231,7 @@ def main(): infile += '/golemfit/' elif args.likelihood is Likelihood.GAUSSIAN: infile += '/gaussian/' + # infile += '/DIM{0}/fix_ifr/HESESim'.format(dim) infile += '/DIM{0}/fix_ifr/'.format(dim) if args.likelihood is Likelihood.GAUSSIAN: infile += '{0}/'.format(str(args.sigma_ratio).replace('.', '_')) @@ -273,6 +278,7 @@ def main(): argsc.dimension = dim _, scale_region = fr_utils.estimate_scale(argsc) argsc.scale_region = scale_region + # infile = base_infile + '/DIM{0}/fix_ifrHESESim/'.format(dim) infile = base_infile + '/DIM{0}/fix_ifr/'.format(dim) if args.likelihood is Likelihood.GAUSSIAN: infile += '{0}/'.format(str(args.sigma_ratio).replace('.', '_')) @@ -282,8 +288,11 @@ def main(): argsc.source_ratio = src finfile = infile +'/{0}/{1}/fr_stat'.format(args.stat_method, args.run_method) \ + misc_utils.gen_identifier(argsc) - basename = os.path.dirname(finfile) + '/statrun/' + os.path.basename(finfile) + basename = os.path.dirname(finfile) + '/statrun/' baseoutfile = basename[:5]+basename[5:].replace('data', 'plots') + if args.data: + baseoutfile += '/data/' + baseoutfile += os.path.basename(finfile) if args.run_method is SensitivityCateg.FULL: outfile = baseoutfile plot_utils.plot_statistic( @@ -331,11 +340,12 @@ def main(): if args.plot: print 'Plotting sensitivities' + basename = args.infile[:5]+args.infile[5:].replace('data', 'plots') if args.likelihood in [Likelihood.GOLEMFIT, Likelihood.GF_FREQ]: - basename = args.infile + 'golemfit/{1}'.format(args.likelihood, args.stat_method) + baseoutfile = basename + 'golemfit/{1}'.format(args.likelihood, args.stat_method) + if args.data: baseoutfile += '/data' elif args.likelihood is Likelihood.GAUSSIAN: - basename = args.infile + 'gaussian/{1}'.format(args.likelihood, args.stat_method) - baseoutfile = basename[:5]+basename[5:].replace('data', 'plots') + baseoutfile = basename + 'gaussian/{1}'.format(args.likelihood, args.stat_method) if args.run_method is SensitivityCateg.FULL: plot_utils.plot_sens_full( -- cgit v1.2.3