src/calibration/__init__.py

"""
Calibration and uncertainty quantification module for healthcare-safe DDI predictions.

Implements:
- Temperature scaling calibration
- Class-wise temperature calibration
- Isotonic regression calibration
- Expected Calibration Error (ECE) calculation
- Reliability diagram generation
- Uncertainty quantification
"""

from .temperature_scaling import (
    TemperatureScaling,
    ClassWiseTemperatureScaling,
    calibrate_logits,
)
from .isotonic_regression import IsotonicRegression
from .calibration_metrics import (
    compute_ece,
    compute_mce,
    compute_brier_score,
    compute_log_loss,
    compute_prediction_variance,
)
from .reliability_analysis import (
    generate_reliability_diagram,
    generate_confidence_histogram,
    generate_confidence_vs_accuracy,
    generate_calibration_summary,
)
from .uncertainty import (
    compute_entropy,
    compute_margin,
    compute_uncertainty,
)
from .escalation_policy import (
    UncertaintyAwareEscalationPolicy,
    EscalationDecision,
)

__all__ = [
    "TemperatureScaling",
    "ClassWiseTemperatureScaling",
    "calibrate_logits",
    "IsotonicRegression",
    "compute_ece",
    "compute_mce",
    "compute_brier_score",
    "compute_log_loss",
    "compute_prediction_variance",
    "generate_reliability_diagram",
    "generate_confidence_histogram",
    "generate_confidence_vs_accuracy",
    "generate_calibration_summary",
    "compute_entropy",
    "compute_margin",
    "compute_uncertainty",
    "UncertaintyAwareEscalationPolicy",
    "EscalationDecision",
]