mlrl.testbed_sklearn.experiments.output.characteristics.data.writer_prediction module

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

Provides classes for writing characteristics of binary predictions to one or several sinks.

class mlrl.testbed_sklearn.experiments.output.characteristics.data.writer_prediction.PredictionCharacteristicsWriter(*extractors: DataExtractor)

Bases: OutputWriter

Allows to write the characteristics of binary predictions to one or several sinks.

class DefaultExtractor

Bases: DataExtractor

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

extract_data(state: ExperimentState, _: List[Sink]) → OutputData | None

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