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 --- utils/gf.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) (limited to 'utils/gf.py') diff --git a/utils/gf.py b/utils/gf.py index b651b5a..7ded152 100644 --- a/utils/gf.py +++ b/utils/gf.py @@ -55,7 +55,8 @@ def fit_flags(llh_paramset): def steering_params(args): steering_categ = args.ast - params = gf.SteeringParams() + # params = gf.SteeringParams(gf.sampleTag.HESE) + params = gf.SteeringParams(gf.sampleTag.MagicTau) params.quiet = False params.fastmode = True params.simToLoad= steering_categ.name.lower() @@ -65,12 +66,15 @@ def steering_params(args): # For Tianlu # params.years = [999] + params.minFitEnergy = 1.0e5 # GeV + return params def set_up_as(fitter, params): print 'Injecting the model', params - asimov_params = gf.FitParameters(gf.sampleTag.HESE) + # asimov_params = gf.FitParameters(gf.sampleTag.HESE) + asimov_params = gf.FitParameters(gf.sampleTag.MagicTau) for parm in params: asimov_params.__setattr__(parm.name, float(parm.value)) fitter.SetupAsimov(asimov_params) @@ -81,12 +85,14 @@ def setup_fitter(args, asimov_paramset): sparams = steering_params(args) npp = gf.NewPhysicsParams() fitter = gf.GolemFit(datapaths, sparams, npp) - set_up_as(fitter, asimov_paramset) + # comment to use data + # set_up_as(fitter, asimov_paramset) return fitter def get_llh(fitter, params): - fitparams = gf.FitParameters(gf.sampleTag.HESE) + # fitparams = gf.FitParameters(gf.sampleTag.HESE) + fitparams = gf.FitParameters(gf.sampleTag.MagicTau) for parm in params: fitparams.__setattr__(parm.name, float(parm.value)) llh = -fitter.EvalLLH(fitparams) @@ -95,7 +101,8 @@ def get_llh(fitter, params): def get_llh_freq(fitter, params): print 'setting to {0}'.format(params) - fitparams = gf.FitParameters(gf.sampleTag.HESE) + # fitparams = gf.FitParameters(gf.sampleTag.HESE) + fitparams = gf.FitParameters(gf.sampleTag.MagicTau) for parm in params: fitparams.__setattr__(parm.name, float(parm.value)) fitter.SetFitParametersSeed(fitparams) -- cgit v1.2.3