swvo.io.RBMDataSet.interp_functions#

Functions

interp_flux(self, target_en, ...], ...)

interp_psd(self, target_mu, ...], ...)

Interpolate PSD to requested (mu, K) targets for every time.

Classes

TargetType(*values)

class swvo.io.RBMDataSet.interp_functions.TargetType(*values)[source]#

Bases: Enum

swvo.io.RBMDataSet.interp_functions.interp_psd(self: <module 'swvo.io.RBMDataSet.RBMDataSet' from '/home/docs/checkouts/readthedocs.org/user_builds/swvo/checkouts/stable/swvo/io/RBMDataSet/RBMDataSet.py'>, target_mu: float | list[float] | ~numpy.ndarray[tuple[~typing.Any, ...], ~numpy.dtype[~numpy.float64]], target_K: float | list[float] | ~numpy.ndarray[tuple[~typing.Any, ...], ~numpy.dtype[~numpy.float64]], target_type: ~swvo.io.RBMDataSet.interp_functions.TargetType | ~typing.Literal['TargetPairs', 'TargetMesh'], n_threads: int = 10) → ndarray[tuple[Any, ...], dtype[float64]][source]#

Interpolate PSD to requested (mu, K) targets for every time.

Output shapes (matching interp_flux semantics):

  • TargetPairs -> (time, N)

  • TargetMeshGrid -> (time, n_mu, n_K)