diff options
Diffstat (limited to 'utils/likelihood.py')
| -rw-r--r-- | utils/likelihood.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/utils/likelihood.py b/utils/likelihood.py index 9bba79a..f590200 100644 --- a/utils/likelihood.py +++ b/utils/likelihood.py @@ -120,6 +120,8 @@ def triangle_llh(theta, args, asimov_paramset, mcmc_paramset, fitter): for idx, param in enumerate(hypo_paramset.from_tag(ParamTag.BESTFIT)): param.value = fr[idx] + # print 'hypo_paramset', hypo_paramset + if args.likelihood is Likelihood.FLAT: return 1. elif args.likelihood is Likelihood.GAUSSIAN: @@ -127,3 +129,12 @@ def triangle_llh(theta, args, asimov_paramset, mcmc_paramset, fitter): return gaussian_llh(fr, fr_bf, args.sigma_ratio) elif args.likelihood is Likelihood.GOLEMFIT: return gf_utils.get_llh(fitter, hypo_paramset) + +def ln_prob(theta, args, fitter, asimov_paramset, mcmc_paramset): + lp = lnprior(theta, paramset=mcmc_paramset) + if not np.isfinite(lp): + return -np.inf + return lp + triangle_llh( + theta, args=args, asimov_paramset=asimov_paramset, + mcmc_paramset=mcmc_paramset, fitter=fitter + ) |