mlrl.boosting.cython.probability_calibration module

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

class mlrl.boosting.cython.probability_calibration.IsotonicJointProbabilityCalibratorConfig

Bases: object

Allows to configure a calibrator that fits a model for the calibration of joint probabilities via isotonic regression.

is_holdout_set_used() → bool

Returns whether the calibration model is fit to the examples in the holdout set, if available, or not.

Returns:

True, if the calibration model is fit to the examples in the holdout set, if available, False if the training set is used instead

set_use_holdout_set(use_holdout_set: bool) → IsotonicJointProbabilityCalibratorConfig

Sets whether the calibration model should be fit to the examples in the holdout set, if available, or not.

Parameters:

use_holdout_set – True, if the calibration model should be fit to the examples in the holdout set, if available, False if the training set should be used instead

Returns:

An IsotonicJointProbabilityCalibratorConfig that allows further configuration of the calibrator

class mlrl.boosting.cython.probability_calibration.IsotonicMarginalProbabilityCalibratorConfig

Bases: object

Allows to configure a calibrator that fits a model for the calibration of marginal probabilities via isotonic regression.

is_holdout_set_used() → bool

Returns whether the calibration model is fit to the examples in the holdout set, if available, or not.

Returns:

True, if the calibration model is fit to the examples in the holdout set, if available, False if the training set is used instead

set_use_holdout_set(use_holdout_set: bool) → IsotonicMarginalProbabilityCalibratorConfig

Sets whether the calibration model should be fit to the examples in the holdout set, if available, or not.

Parameters:

use_holdout_set – True, if the calibration model should be fit to the examples in the holdout set, if available, False if the training set should be used instead

Returns:

An IsotonicMarginalProbabilityCalibratorConfig that allows further configuration of the calibrator