diff options
Diffstat (limited to 'misc')
| -rw-r--r-- | misc/austin.py | 154 | ||||
| -rw-r--r-- | misc/command | 8 | ||||
| -rw-r--r-- | misc/nufit_u.txt | 14 |
3 files changed, 0 insertions, 176 deletions
diff --git a/misc/austin.py b/misc/austin.py deleted file mode 100644 index dddab30..0000000 --- a/misc/austin.py +++ /dev/null @@ -1,154 +0,0 @@ -#! /usr/bin/env python -# author : S. Mandalia -# s.p.mandalia@qmul.ac.uk -# -# date : February 24, 2019 - -""" -HESE BSM Flavour Figure 2 -""" - -from __future__ import absolute_import, division - -import argparse -from functools import partial - -import numpy as np - -from golemflavor import fr as fr_utils -from golemflavor import misc as misc_utils -from golemflavor import plot as plot_utils -from golemflavor.enums import str_enum -from golemflavor.enums import Likelihood, ParamTag, PriorsCateg -from golemflavor.param import Param, ParamSet - -from matplotlib import pyplot as plt - -# from pymultinest import Analyzer -import json - - -def define_nuisance(): - """Define the nuisance parameters.""" - nuisance = [] - tag = ParamTag.NUISANCE - lg_prior = PriorsCateg.LIMITEDGAUSS - nuisance.extend([ - Param(name='convNorm', value=1., seed=[0.5, 2. ], ranges=[0.1, 10.], std=0.4, prior=lg_prior, tag=tag), - Param(name='promptNorm', value=0., seed=[0. , 6. ], ranges=[0. , 20.], std=2.4, prior=lg_prior, tag=tag), - Param(name='muonNorm', value=1., seed=[0.1, 2. ], ranges=[0. , 10.], std=0.1, tag=tag), - Param(name='astroNorm', value=6.9, seed=[0., 5. ], ranges=[0. , 20.], std=1.5, tag=tag), - Param(name='astroDeltaGamma', value=2.5, seed=[2.4, 3. ], ranges=[-5., 5. ], std=0.1, tag=tag) - ]) - return ParamSet(nuisance) - - -def get_paramsets(args, nuisance_paramset): - paramset = [] - if args.likelihood in [Likelihood.GOLEMFIT, Likelihood.GF_FREQ]: - gf_nuisance = [x for x in nuisance_paramset.from_tag(ParamTag.NUISANCE)] - paramset.extend(gf_nuisance) - tag = ParamTag.BESTFIT - paramset.extend([ - Param(name='astroFlavorAngle1', value=0, ranges=[0., 1.], std=0.2, tag=tag), - Param(name='astroFlavorAngle2', value=0, ranges=[-1., 1.], std=0.2, tag=tag), - ]) - paramset = ParamSet(paramset) - return paramset - - -def process_args(args): - """Process the input args.""" - if args.likelihood is not Likelihood.GOLEMFIT \ - and args.likelihood is not Likelihood.GF_FREQ: - raise AssertionError( - 'Likelihood method {0} not supported for this ' - 'script!\nChoose either GOLEMFIT or GF_FREQ'.format( - str_enum(args.likelihood) - ) - ) - -def parse_args(args=None): - """Parse command line arguments""" - parser = argparse.ArgumentParser( - description="HESE BSM Flavour Figure 2", - formatter_class=misc_utils.SortingHelpFormatter, - ) - parser.add_argument( - '--likelihood', default='golemfit', - type=partial(misc_utils.enum_parse, c=Likelihood), - choices=Likelihood, help='likelihood contour' - ) - parser.add_argument( - '--contour-dir', type=str, - help='Path to directory containing MultiNest runs' - ) - parser.add_argument( - '--outfile', type=str, default='./untitled', - help='Output path' - ) - if args is None: return parser.parse_args() - else: return parser.parse_args(args.split()) - - -def main(): - args = parse_args() - process_args(args) - misc_utils.print_args(args) - - paramset = get_paramsets(args, define_nuisance()) - n_params = len(paramset) - print n_params - - # Data - prefix = 'austin' - # data_path = '/home/aschneider/programs/GOLEMSPACE/sources/GolemFit/scripts/diffuse/mcmcs/results/dpl_numu_prior_flavor_20190302-162221-a747f528-8aa6-4488-8c80-059572c099fe.json' - # data_path = '/home/aschneider/programs/GOLEMSPACE/sources/GolemFit/scripts/diffuse/mcmcs/results/spl_flavor_20190311-170924-5297d736-3c6e-447f-8de7-4a0653a51bb6.json' - data_path = '/home/aschneider/programs/GOLEMSPACE/sources/GolemFit/scripts/diffuse/mcmcs/results/spl_flavor_20190420-161513-524f1731-0bcb-49e3-a2ea-ff3c69b4e53c.json' - with open(data_path) as f: - d_json = json.load(f) - names = d_json['func_args'] - chains = np.array(d_json['chain']) - print 'names', names - print 'chains.shape', chains.shape - flavour_angles = chains[:,4:6] - flavour_ratios = np.array( - map(fr_utils.angles_to_fr, flavour_angles) - ) - - # # Load HESE contour. - # prefix = 'shivesh' - # contour_infile = '/data/user/smandalia/flavour_ratio/data/contour/contour_REAL.npy' - # contour_angles = np.load(contour_infile)[:,-2:] - # flavour_ratios = np.array(map(fr_utils.angles_to_fr, contour_angles)) - - nbins = 25 - ax_labels = [r'$f_{e}^{\oplus}$', r'$f_{\mu}^{\oplus}$', r'$f_{\tau}^{\oplus}$'] - - fig = plt.figure(figsize=(8, 8)) - ax = fig.add_subplot(111) - tax = plot_utils.get_tax(ax, scale=nbins, ax_labels=ax_labels) - - levels = [10, 20, 40, 60, 68, 80, 90, 99] - for l in levels: - plot_utils.flavour_contour( - frs = flavour_ratios, - ax = ax, - nbins = nbins, - coverage = l, - linewidth = 2, - # color = 'green' - ) - - ax.legend() - - fig.savefig('contour_{0}.png'.format(prefix), bbox_inches='tight', dpi=150) - - print "DONE!" - - -main.__doc__ = __doc__ - - -if __name__ == '__main__': - main() diff --git a/misc/command b/misc/command deleted file mode 100644 index 3ec2d6c..0000000 --- a/misc/command +++ /dev/null @@ -1,8 +0,0 @@ -python contour.py --data real --debug true --nsteps 100 --burnin 10 --nwalkers 20 --datadir ./test_ --seed 26 --threads max -python sens.py --debug True --data real --datadir ./test --dimension 6 --eval-segment 1 --mn-live-points 100 --mn-output ./test --mn-tolerance 0.3 --seed 26 --segments 10 --source-ratio 1 2 0 --stat-method bayesian --threads 4 --texture oeu - -python plot_sens.py --data real --datadir /data/user/smandalia/flavour_ratio/data/sensitivity/ --dimensions 6 --plot-x True --segments 10 --x-segments 20 --split-jobs True --stat-method bayesian --texture none -python mc_unitary.py --burnin 100 --datadir=/data/user/smandalia/flavour_ratio/data/mc_unitary --nsteps 1000 --nwalkers 60 --run-mcmc True --seed 26 --source-ratio 1 2 0 --threads 1 -python mc_x.py --burnin 200 --datadir=/data/user/smandalia/flavour_ratio/data/mc_x --nsteps 1000 --nwalkers 60 --seed 26 --threads max -python fr.py --data real --datadir ./test/ --dimension 6 --nsteps 20 --nwalkers 60 --burnin 20 --seed 26 --source-ratio 0 1 0 --texture oet --threads max --debug True -python plot_sens.py --data real --datadir /data/user/smandalia/flavour_ratio/data/sensitivity/ --dimensions 3 4 5 6 7 8 --plot-table True --segments 10 --source-ratios 0 1 0 1 0 0 --split-jobs True --stat-method bayesian --plot-statistic False diff --git a/misc/nufit_u.txt b/misc/nufit_u.txt deleted file mode 100644 index 4e7433e..0000000 --- a/misc/nufit_u.txt +++ /dev/null @@ -1,14 +0,0 @@ - -[[ 0.82327921+0.j 0.54796108+0.j -0.09913534+0.11010079j] - [-0.30340559+0.06889398j 0.59033699+0.0458547j 0.74336952+0.j ] - [ 0.47090947+0.06045075j -0.58950774+0.04023502j 0.65226662+0.j ]] - -abs: -[[ 0.82327921 0.54796108 0.14815532] - [ 0.31112912 0.59211521 0.74336952] - [ 0.47477365 0.5908792 0.65226662]] - -M -[[0, 0, 0 ], - [0, 7.40E-23, 0 ], - [0, 0, 2.515E-21]] |