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: Extension

An extension that configures the functionality to split tabular datasets into training and test datasets.

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

static get_random_state(args: Namespace) int

Returns the seed to be used by random number generators.

Parameters:

args – The command line arguments specified by the user

Returns:

The seed to be used