DataModule.from_dataset

classmethod DataModule.from_dataset(dataset: Dataset, input_ml_names: List[str] | None = None, output_ml_names: List[str] | None = None, input_ml_transformation: DataBlockTransformation | None = None, output_ml_transformation: DataBlockTransformation | None = None, **kwargs)[source]

Creates a data module from a dataset.

Parameters:

  • dataset (Dataset) – The dataset to be used to create the data module.

  • input_ml_names (List[str], optional) – List of names of the input data to be used for the ML model. The default is None. If None, then all the design parameters are used.

  • output_ml_names (List[str], optional) – List of names of the output data to be used for the ML model. The default is None. If None, then all the performance attributes are used.

  • input_ml_transformation (DataBlockTransformation, optional) – Custom transformation to be used for the input data. The default is None.

  • output_ml_transformation (DataBlockTransformation, optional) – Custom transformation to be used for the output data. The default is None.

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

Returns:

DataModule – The data module created from the dataset.