Source code for caliber.multiclass_classification.minimizing.linear_scaling.calibration.brier_linear_scaling
from typing import Optional
from caliber.multiclass_classification.metrics import brier_score_loss
from caliber.multiclass_classification.minimizing.linear_scaling.calibration.base import (
CalibrationLinearScalingMulticlassClassificationModel,
)
from caliber.multiclass_classification.minimizing.linear_scaling.linear_scaling_smooth_fit_mixin_ import (
LinearScalingSmoothFitMulticlassClassificationMixin,
)
[docs]
class BrierLinearScalingMulticlassClassificationModel(
LinearScalingSmoothFitMulticlassClassificationMixin,
CalibrationLinearScalingMulticlassClassificationModel,
):
def __init__(
self,
minimize_options: Optional[dict] = None,
has_intercept: bool = True,
has_shared_intercept: bool = False,
has_cross_slopes: bool = True,
has_shared_slope: bool = False,
):
super().__init__(
brier_score_loss,
minimize_options,
has_intercept=has_intercept,
has_shared_intercept=has_shared_intercept,
has_shared_slope=has_shared_slope,
has_cross_slopes=has_cross_slopes,
)