mlrl.testbed.experiments.input.dataset.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.extension.DatasetExtension(*dependencies: Extension)

Bases: Extension, ABC

An abstract base class for all extensions that configure the functionality to load datasets.

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

get_supported_modes() → Set[Type[Mode]]

See mlrl.testbed.extensions.extension.Extension.get_supported_modes()

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

Bases: DatasetExtension, ABC

An abstract base class for all extensions that configure the functionality to load datasets from files.

DATASET_DIRECTORY = <mlrl.util.cli.StringArgument object>

static parse_dataset_args_from_config(config: ConfigFile) → List[ArgumentList]

Parses and returns the command line arguments for using the datasets specified in a configuration file.

Parameters:

config – The configuration file that should be parsed

Returns:

A list that contains the command line arguments that have been parsed