Source code for caliber.binary_classification.minimizing.linear_scaling.calibration.beta
from typing import Optional
from caliber.binary_classification.minimizing.linear_scaling.calibration.cross_entropy_linear_scaling import (
CrossEntropyLinearScalingBinaryClassificationModel,
)
[docs]
class BetaBinaryClassificationModel(CrossEntropyLinearScalingBinaryClassificationModel):
def __init__(
self,
minimize_options: Optional[dict] = None,
):
super().__init__(
minimize_options=minimize_options,
has_intercept=True,
has_bivariate_slope=True,
)
[docs]
class DiagBetaBinaryClassificationModel(
CrossEntropyLinearScalingBinaryClassificationModel
):
def __init__(
self,
minimize_options: Optional[dict] = None,
):
super().__init__(
minimize_options=minimize_options,
has_intercept=True,
has_bivariate_slope=False,
)