Skip to content

Examples for the use of NABU to model and communicate an unbinned likelihood from samples

Notifications You must be signed in to change notification settings

nabu-hep/examples

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Examples for Nabu

arxiv DOI

This repository contains the likelihood models presented in the paper and instructions on replicating the results using the nabu package. Below, we describe how to run each analysis. The following prescription allows each model to be loaded and used.

import nabu
likelihood_model = nabu.Likelihood.load("PATH/TO/MODEL.nabu")
generated_samples = likelihood_model.sample(100)

where the last line shows how to generate samples from the likelihood model.

The ATLAS model utilised in the paper is included in ATLAS/atlas-model.nabu, which encompasses the transformation information. To replicate the results, one can use the following command:

nabu-fit-to-data -dp ATLAS/atlas-dataset.npz -e 600 -t rqs -k 12 -int 1e-6 1 -tfrac 0.2990332256 -w 512 -l 8 -d 1 -perm random -lr 0.01 -mlr 1.0e-06

The data file contains standardised version of $p_T^{\mu\mu}$, $y^{\mu\mu}$, $p_T^{\mu_1}$, $\Delta \eta_{12}^2$ and $\Delta \phi_{12}^2$, respectively.

The LHCb model used in the paper is located in LHCb/lhcb-model.nabu, which includes the transformation information. To replicate the results, use the following command.

nabu-fit-to-data -dp LHCb/lhcb-dataset.npz -e 600 -t rqs -k 12 -int 1e-6 1 -w 64 -l 8 -d 2 -lr 0.001 -mlr 1.0e-08

The datafile contains standardised version of $m^2(K^+D^-)$ and $m^2(D^+D^-)$, respectively.

The WET model used in this paper is included in WET/wet-model.nabu, which contains the transformation information. To replicate the results, one can use the following command.

nabu-fit-to-data -dp WET/wet-dataset.npz -e 600 -w 64 -l 2 -d 3 -lr 0.001 -mlr 0.0001

The datafile contains standardised version of ${\rm Re}[C_{V_R}^{\bar{u}b\bar{e}\nu_e}]$, ${\rm Re}[C_{V_L}^{\bar{u}b\bar{e}\nu_e}]$, ${\rm Re}[C_{S_R}^{\bar{u}b\bar{e}\nu_e}]$, ${\rm Re}[C_{S_L}^{\bar{u}b\bar{e}\nu_e}]$ and ${\rm Re}[C_{T}^{\bar{u}b\bar{e}\nu_e}]$, respectively.

Sample illustrations

The complexity of the different samples can be illustrated using

python visualisation.py {ATLAS,LHCb,WET}

Please note that in addition to the packages required in the main repo (see here), the script uses pandas, mplhep, and hist which are available with pip and conda.

Figure Caption
One- and two-dimensional projections of the ATLAS example data set.
One- and two-dimensional projections of the LHCb example data set.
One- and two-dimensional projections of the WET example data set.

About

Examples for the use of NABU to model and communicate an unbinned likelihood from samples

Resources

Stars

Watchers

Forks

Contributors 3

  •  
  •  
  •  

Languages