mlrl.testbed.experiments.input.dataset package

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

Provides classes that allow to read datasets from different sources.

class mlrl.testbed.experiments.input.dataset.DatasetReader(input_data: InputDataset, *sources: Source)

Bases: InputReader

Allows reading a dataset from one or several sources.

add_preprocessors(*preprocessors: Preprocessor)

Adds one or several preprocessors that should be applied to loaded datasets.

Parameters:

preprocessors – The preprocessors to be added

Returns:

The DatasetReader itself

read(state: ExperimentState) → ExperimentState

See mlrl.testbed.experiments.input.reader.InputReader.read()

class mlrl.testbed.experiments.input.dataset.InputDataset(name: str)

Bases: DatasetInputData

Represents a dataset that is part of input data.

NAME = 'Ground truth'
update_state(state: ExperimentState, input_data: Any)

See mlrl.testbed.experiments.input.data.InputData.update_state()

Subpackages

Submodules