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: Union[float, numpy.ndarray, torch.Tensor]) → Union[float, numpy.ndarray, torch.Tensor]

Scale values.

Parameters

x – Values to be scaled.

Returns

Scaled values.

abstract transform_value_inv(x: Union[float, numpy.ndarray, torch.Tensor]) → Union[float, numpy.ndarray, torch.Tensor]

Invert scaling of values.

Parameters

x – Values where scaling should be inverted.

Returns

Values with inverse scaling.