pytorch_forecasting.data.timeseries.check_for_nonfinite

pytorch_forecasting.data.timeseries.check_for_nonfinite(tensor: torch.Tensor, names: Union[str, List[str]]) → torch.Tensor

Check if 2D tensor contains NAs or inifinite values.

Parameters

• names (Union[str, List[str]]) – name(s) of column(s) (used for error messages)

• tensor (torch.Tensor) – tensor to check

Returns

returns tensor if checks yield no issues

Return type

torch.Tensor

pytorch_forecasting.data.timeseries._find_end_indices TimeSeriesDataSet