Add ConformalRiskCalibrator and ConformalRiskPredictor for conformal risk control prediction sets

Signed-off-by: Colin Son <txmed82@users.noreply.github.com>

Author: txmed82

docs/source/metrics.rst

@@ -203,6 +203,26 @@ Metrics
     :members:
 
 
+`Conformal Risk Control`
+------------------------
+.. autoclass:: ConformalRiskCalibrator
+    :members:
+
+.. autoclass:: ConformalRiskPredictor
+    :members:
+    :special-members: __call__
+
+.. autoclass:: Coverage
+    :members:
+
+.. autoclass:: SetSize
+    :members:
+
+.. autofunction:: compute_coverage
+
+.. autofunction:: compute_set_size
+
+
 Utilities
 ---------
 .. automodule:: monai.metrics.utils

monai/metrics/__init__.py

@@ -14,6 +14,14 @@
 from .active_learning_metrics import LabelQualityScore, VarianceMetric, compute_variance, label_quality_score
 from .average_precision import AveragePrecisionMetric, compute_average_precision
 from .calibration import CalibrationErrorMetric, CalibrationReduction, calibration_binning
+from .conformal_risk import (
+    ConformalRiskCalibrator,
+    ConformalRiskPredictor,
+    Coverage,
+    SetSize,
+    compute_coverage,
+    compute_set_size,
+)
 from .confusion_matrix import ConfusionMatrixMetric, compute_confusion_matrix_metric, get_confusion_matrix
 from .cumulative_average import CumulativeAverage
 from .embedding_collapse import EmbeddingCollapseMetric, compute_embedding_collapse