DataModule.normalize_y

DataModule.normalize_y(y: torch.Tensor | ndarray) → torch.Tensor | ndarray

Normalize only the output data. Takes care of the case where y is a numpy array or a torch tensor.

Parameters:

y (Union[torch.Tensor, np.ndarray]) – A numpy array or a torch tensor to be normalized.

Returns:

Union[torch.Tensor, np.ndarray] – The normalized data. The output has the same type as the input.