mlrl.common.testbed.experiments.output.label_vectors.extension module

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

Provides classes that allow configuring the functionality to write label vectors to one or several sinks.

class mlrl.common.testbed.experiments.output.label_vectors.extension.LabelVectorSetExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the functionality to write label vectors, extracted from a LabelVectorSet, to one or several sinks.

class LabelVectorSetExtractor

Bases: DataExtractor

Allows to extract unique label vectors from a LabelVectorSet.

class Visitor

Bases: LabelVectorSetVisitor

Accesses the label vectors and frequencies stored by a LabelVectorSet and stores them in a LabelVectorHistogram.

visit_label_vector(label_vector: ndarray, frequency: int)

See mlrl.common.cython.output_space_info.LabelVectorSetVisitor.visit_label_vector()

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

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

configure_experiment(args: Namespace, experiment_builder: Builder, mode: ExperimentMode)

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

get_supported_modes() → set[ExperimentMode]

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