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.

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

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

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>