DataModule.from_parameters

classmethod DataModule.from_parameters(x: ndarray | torch.Tensor | None = None, y: ndarray | torch.Tensor | None = None, **datamodule_kwargs)[source]

Creates a data module from parameters returned by the get_parameters(…) method.

Parameters:
  • x (Union[np.ndarray, torch.Tensor, None]) – The input data.

  • y (Union[np.ndarray, torch.Tensor, None]) – The output data.

  • **datamodule_kwargs – Additional keyword arguments to be passed to the data module.

Returns:

DataModule – The data module created from the parameters.