ValueTransform

class maze.train.trainers.common.value_transform.ValueTransform

Value transformation (e.g. useful for training the critic in Alpha(Mu)Zero).

abstract transform_value(x: float | numpy.ndarray | torch.Tensor) → float | numpy.ndarray | torch.Tensor

Scale values.

Parameters:

x – Values to be scaled.

Returns:

Scaled values.

abstract transform_value_inv(x: float | numpy.ndarray | torch.Tensor) → float | numpy.ndarray | torch.Tensor

Invert scaling of values.

Parameters:

x – Values where scaling should be inverted.

Returns:

Values with inverse scaling.