data

import tensorwaves.data

The data module takes care of data generation.

generate_data(size: int, initial_state_mass: float, final_state_masses: Mapping[int, float], data_transformer: DataTransformer, intensity: Function, phsp_generator: Optional[PhaseSpaceGenerator] = None, random_generator: Optional[UniformRealNumberGenerator] = None, bunch_size: int = 50000) Mapping[Union[int, str], ndarray][source]

Facade function for creating data samples based on an intensities.

Parameters
  • size – Sample size to generate.

  • initial_state_mass – See PhaseSpaceGenerator.setup().

  • final_state_masses – See PhaseSpaceGenerator.setup().

  • data_transformer – An instance of DataTransformer that is used to transform a generated DataSample to a DataSample that can be understood by the Function.

  • intensity – The intensity Function that will be sampled.

  • phsp_generator – Class of a phase space generator.

  • random_generator – A uniform real random number generator. Defaults to TFUniformRealNumberGenerator with indeterministic behavior.

  • bunch_size – Adjusts size of a bunch. The requested sample size is generated from many smaller samples, aka bunches.

generate_phsp(size: int, initial_state_mass: float, final_state_masses: Mapping[int, float], phsp_generator: Optional[PhaseSpaceGenerator] = None, random_generator: Optional[UniformRealNumberGenerator] = None, bunch_size: int = 50000) Mapping[Union[int, str], ndarray][source]

Facade function for creating (unweighted) phase space samples.

Parameters

Submodules and Subpackages