mlrl.testbed_sklearn.experiments.output.dataset.writer_ground_truth module¶

Provides classes that allow writing the ground truth to one or several sinks.

class mlrl.testbed_sklearn.experiments.output.dataset.writer_ground_truth.GroundTruthWriter(*extractors: DataExtractor)¶

Allows to write the ground truth for tabular data to one or several sinks.

class DefaultExtractor¶

Bases: DataExtractor

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

extract_data(state: ExperimentState, _: List[Sink]) → OutputData | None¶: See mlrl.testbed.experiments.output.writer.DataExtractor.extract_data()