mlrl.testbed.experiments.input.dataset.splitters.arguments module¶

Author: Michael Rapp (michael.rapp.ml@gmail.com)

Defines command line arguments for configuring the functionality to split datasets into training and test datasets.

class mlrl.testbed.experiments.input.dataset.splitters.arguments.DatasetSplitterArguments¶

Bases: object

Defines command line arguments for configuring the functionality to split datasets into training and test datasets.

DATASET_SPLITTER = <mlrl.util.cli.SetArgument object>¶

OPTION_FIRST_FOLD = 'first_fold'¶

OPTION_LAST_FOLD = 'last_fold'¶

OPTION_NUM_FOLDS = 'num_folds'¶

OPTION_TEST_SIZE = 'test_size'¶

RANDOM_STATE = <mlrl.util.cli.IntArgument object>¶

VALUE_CROSS_VALIDATION = 'cross-validation'¶

VALUE_TRAIN_TEST = 'train-test'¶