mlrl.testbed.experiments.output.parameters.writer module

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

Provides classes that allow writing algorithmic parameters to one or several sinks.

class mlrl.testbed.experiments.output.parameters.writer.ParameterWriter(*extractors: DataExtractor)

Bases: OutputWriter

Allows writing algorithmic parameters to one or several sinks.

class DefaultExtractor

Bases: DataExtractor

The extractor to be used by a ParameterWriter, by default.

extract_data(state: ExperimentState, _: list[Sink]) → list[tuple[ExperimentState, OutputData]]

See mlrl.testbed.experiments.output.writer.DataExtractor.extract_data()

create_input_reader(args: Namespace, input_directory: Path) → InputReader | None

May be overridden by subclasses in order to create an InputReader that can read the data produced by this output writer.

Parameters:

• args – The command line arguments specified by the user

• input_directory – The directory, the data should be read from

Returns:

The InputReader that has been created or None, if no such reader is available