mlrl.common.testbed.experiments.output.characteristics.model.writer module

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

Provides classes that allow writing characteristics of models to one or several sinks.

class mlrl.common.testbed.experiments.output.characteristics.model.writer.RuleModelCharacteristicsWriter(*extractors: DataExtractor)

Bases: ResultWriter

Allows writing the characteristics of a model to one or several sinks.

class DefaultExtractor

Bases: DataExtractor

Allows to extract characteristics from a RuleModel.

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

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

class InputExtractor(properties: TabularProperties, context: Context)

Bases: TabularDataExtractor

Uses TabularInputData that has previously been loaded via an input reader.

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

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