nami_toys.gaussian

Classes

GaussianMixture([sig_loc, sig_cov, bkg_loc, ...])

N-dimensional Gaussian signal + background simulator.

class nami_toys.gaussian.GaussianMixture(sig_loc=<factory>, sig_cov=<factory>, bkg_loc=<factory>, bkg_cov=<factory>)

Bases: object

N-dimensional Gaussian signal + background simulator.

Parameters:

• sig_loc (Tensor) – Mean (d,) and covariance (d, d) of the signal component.

• sig_cov (Tensor) – Mean (d,) and covariance (d, d) of the signal component.

• bkg_loc (Tensor) – Mean (d,) and covariance (d, d) of the background component.

• bkg_cov (Tensor) – Mean (d,) and covariance (d, d) of the background component.

property bkg: MultivariateNormal

bkg_cov: Tensor

bkg_loc: Tensor

property d: int

generate(n_expected, sig_frac, *, generator=None)

Draw a Poisson-fluctuated dataset of signal + background events.

Return type:

ToyDataset

Parameters:

• n_expected (int)

• sig_frac (float)

• generator (Generator | None)

property sig: MultivariateNormal

sig_cov: Tensor

sig_loc: Tensor