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

Provides classes that allow writing predictions to one or several sinks.

class mlrl.testbed_sklearn.experiments.output.dataset.writer_prediction.PredictionWriter(*extractors: DataExtractor)¶

Allows to write predictions to one or several sinks.

class DefaultExtractor¶

Bases: DataExtractor

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

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

class PredictionExtractor(properties: Properties, context: Context)¶

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