aboutsummaryrefslogtreecommitdiffstats
path: root/utils/mcmc.py
diff options
context:
space:
mode:
authorshivesh <s.p.mandalia@qmul.ac.uk>2018-04-22 23:18:44 -0500
committershivesh <s.p.mandalia@qmul.ac.uk>2018-04-22 23:18:44 -0500
commit2ca0c5597590e2043bd280dd8aee3d9d09bae29a (patch)
treef1f82bec4213eff4a0d6d8234d2d29cb51f08c72 /utils/mcmc.py
parent7a2920a6fba7a5ef4840785e427995f0b8df0bcc (diff)
downloadGolemFlavor-2ca0c5597590e2043bd280dd8aee3d9d09bae29a.tar.gz
GolemFlavor-2ca0c5597590e2043bd280dd8aee3d9d09bae29a.zip
Sun Apr 22 23:18:44 CDT 2018
Diffstat (limited to 'utils/mcmc.py')
-rw-r--r--utils/mcmc.py9
1 files changed, 8 insertions, 1 deletions
diff --git a/utils/mcmc.py b/utils/mcmc.py
index c712c3a..97b77dd 100644
--- a/utils/mcmc.py
+++ b/utils/mcmc.py
@@ -9,7 +9,6 @@ Useful functions to use an MCMC for the BSM flavour ratio analysis
from __future__ import absolute_import, division
-import argparse
from functools import partial
import emcee
@@ -72,6 +71,14 @@ def mcmc_argparse(parser):
type=partial(enum_parse, c=MCMCSeedType), choices=MCMCSeedType,
help='Type of distrbution to make the initial MCMC seed'
)
+ parser.add_argument(
+ '--plot-angles', type=misc_utils.parse_bool, default='True',
+ help='Plot MCMC triangle in the angles space'
+ )
+ parser.add_argument(
+ '--plot-elements', type=misc_utils.parse_bool, default='False',
+ help='Plot MCMC triangle in the mixing elements space'
+ )
def flat_seed(paramset, nwalkers):