mlrl.testbed.experiments.input.dataset.splitters.extension module

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

Provides classes that allow configuring the functionality to load datasets.

class mlrl.testbed.experiments.input.dataset.splitters.extension.DatasetFileExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the functionality to load datasets from files.

DATASET_DIRECTORY = <mlrl.util.cli.StringArgument object>
DATASET_NAME = <mlrl.util.cli.StringArgument object>
static get_dataset_reader(args: Namespace) DatasetReader

Returns the DatasetReader to be used for loading datasets according to the configuration.

Parameters:

args – The command line arguments specified by the user

Returns:

The DatasetReader to be used