boax.core.models.transformations.transformed

Contents

boax.core.models.transformations.transformed#

boax.core.models.transformations.transformed(model, *, input_transformation_fn=None, outcome_transformation_fn=None)#

Constructs an model with input_transformation_fn applied to its inputs and the output_transformation_fn applied to its outcomes.

Example

>>> transformed = transformed(
  model,
  input_transformation_fn=fn1,
  outcome_transformation_fn=fn2
)
>>> result = transformed(xs)
Parameters:

  • model (Model[TypeVar(A)]) – The base model.

  • input_transformation_fn (InputTransformation | None) – The transformation functions applied to the model’s inputs.

  • outcome_transformation_fn (Optional[OutcomeTransformation[TypeVar(A), TypeVar(B)]]) – The transformation functions applied to the model’s outcomes.

Return type:

Model[TypeVar(B)]

Returns:

The transformed Model function.

Contents