credit.preblock.sqrt

credit.preblock.sqrt#

Classes#

SqrtTransform

Applies a sqrt transformation to specified variables in a batch dict.

Module Contents#

class credit.preblock.sqrt.SqrtTransform(variables: list[str], data_types: list[str] = None)#

Bases: credit.preblock.base.BasePreblock

Applies a sqrt transformation to specified variables in a batch dict.

Expected dict structure:

batch[source][data_type][‘source/var_type/var_shape/var_name’]

Config example:

type: “sqrt_transform” args:

variables:

  • ‘ERA5/prognostic/3D/Q’

data_types: # optional, defaults to [‘input’, ‘target’]

  • ‘input’

  • ‘target’

variables#
data_types = ['input', 'target']#
forward(batch: dict) → dict#