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