prepare_train_batch

prepare_train_batch(data: List[Dict[str, Any]], input_size: int, output_size: int, window_sampling_limit: Optional[int] = None, random_state: Optional[numpy.random.mtrand.RandomState] = None) → Dict[str, Optional[torch.Tensor]]

Prepare batch with training data.

Parameters

• data (List[Dict[str, Any]]) –

• input_size (int) –

• output_size (int) –

• window_sampling_limit (Optional[int]) –

• random_state (Optional[numpy.random.mtrand.RandomState]) –

Return type

Dict[str, Optional[torch.Tensor]]