mlrl.boosting.testbed.experiments.output.probability_calibration package

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

Provides classes that allow to write textual representation of probability calibration models to different sinks.

class mlrl.boosting.testbed.experiments.output.probability_calibration.JointProbabilityCalibrationModelExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the functionality to write models for the calibration of joint probabilities to outputs.

PRINT_JOINT_PROBABILITY_CALIBRATION_MODEL = <mlrl.util.cli.BoolArgument object>
SAVE_JOINT_PROBABILITY_CALIBRATION_MODEL = <mlrl.util.cli.BoolArgument object>
configure_experiment(args: Namespace, experiment_builder: Builder, _: ExperimentMode)

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

get_supported_modes() → set[ExperimentMode]

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

class mlrl.boosting.testbed.experiments.output.probability_calibration.MarginalProbabilityCalibrationModelExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the functionality to write models for the calibration of marginal probabilities to outputs.

PRINT_MARGINAL_PROBABILITY_CALIBRATION_MODEL = <mlrl.util.cli.BoolArgument object>
SAVE_MARGINAL_PROBABILITY_CALIBRATION_MODEL = <mlrl.util.cli.BoolArgument object>
configure_experiment(args: Namespace, experiment_builder: Builder, _: ExperimentMode)

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

get_supported_modes() → set[ExperimentMode]

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

Submodules