Parameter Transformations

class pypomp.core.par_trans.ParTrans(to_est: Callable[[Annotated[dict[str, float | Array], 'ParamDict']], Annotated[dict[str, float | Array], 'ParamDict']] | None = None, from_est: Callable[[Annotated[dict[str, float | Array], 'ParamDict']], Annotated[dict[str, float | Array], 'ParamDict']] | None = None)[source]

Bases: object

Handles parameter transformations between natural and estimation parameter spaces.

Attributes

to_est

The parameter transformation function to the estimation parameter space.

from_est

The parameter transformation function from the estimation parameter space to the natural parameter space.

Methods

panel_transform(theta, direction)

Transform shared and unit-specific parameters for a single replicate.

panel_transform_list(theta_list, direction)

Apply transform to a list of parameter sets.

to_floats(theta, direction)

Convert the theta dictionary values from jax.Array to float.

transform_array(param_array, param_names, ...)

Transform a parameter array to or from the (unconstrained) estimation parameter space.

transform_panel_traces(shared_traces, ...)

Transform panel traces from estimation space to natural space.