mlrl.testbed_sklearn.experiments.input.dataset.splitters.extension module¶
Author: Michael Rapp (michael.rapp.ml@gmail.com)
Provides classes that allow configuring the functionality to split datasets into training and test datasets.
- class mlrl.testbed_sklearn.experiments.input.dataset.splitters.extension.DatasetSplitterExtension(*dependencies: Extension)¶
Bases:
ExtensionAn extension that configures the functionality to split tabular datasets into training and test datasets.
- DATASET_SPLITTER = <mlrl.util.cli.SetArgument object>¶
- RANDOM_STATE = <mlrl.util.cli.IntArgument object>¶
- static get_dataset_splitter(args: Namespace) DatasetSplitter¶
Returns the DatasetSplitter to be used for splitting datasets into training and test datasets according to the configuration.
- Parameters:
args – The command line arguments specified by the user
- Returns:
The DatasetSplitter to be used