Removing the scale from targets

metatrain.utils.scaler.remove.remove_scale(systems: List[System], targets: Dict[str, TensorMap], scaler: Module) → Dict[str, TensorMap]

Scale all targets to a standard deviation of one.

Parameters:

• systems (List[System]) – List of systems corresponding to the targets.

• targets (Dict[str, TensorMap]) – Dictionary containing the targets to be scaled.

• scaler (Module) – The scaler used to scale the targets.

Returns:

The scaled targets.

Return type:

Dict[str, TensorMap]

metatrain.utils.scaler.remove.get_remove_scale_transform(scaler: Scaler) → Callable

Remove the scaling from the targets using the provided scaler.

Parameters:

scaler (Scaler) – The scaler used to scale the targets.

Returns:

A function that removes the scaling from the targets.

Return type:

Callable