9.3. data#

create_data_transformer(model: AmplitudeModel, backend: str = 'jax') SympyDataTransformer[source]#
create_phase_space_filter(decay: ThreeBodyDecay, x_mandelstam: Literal[1, 2, 3] = 1, y_mandelstam: Literal[1, 2, 3] = 2, outside_value=nan) PositionalArgumentFunction[source]#
generate_meshgrid_sample(decay: ThreeBodyDecay, resolution: int, x_mandelstam: Literal[1, 2, 3] = 1, y_mandelstam: Literal[1, 2, 3] = 2) Dict[str, ndarray][source]#

Generate a numpy.meshgrid sample for plotting with matplotlib.pyplot.

generate_sub_meshgrid_sample(decay: ThreeBodyDecay, resolution: int, x_range: tuple[float, float], y_range: tuple[float, float], x_mandelstam: Literal[1, 2, 3] = 1, y_mandelstam: Literal[1, 2, 3] = 2) DataSample[source]#
generate_phasespace_sample(decay: ThreeBodyDecay, n_events: int, seed: int | None = None) DataSample[source]#

Generate a uniform distribution over Dalitz variables \(\sigma_{1,2,3}\).

compute_dalitz_boundaries(decay: ThreeBodyDecay) tuple[tuple[float, float], tuple[float, float], tuple[float, float]][source]#
create_mass_symbol_mapping(decay: ThreeBodyDecay) dict[Symbol, float][source]#