incerto.conformal.prediction_set_from_scores

incerto.conformal.prediction_set_from_scores#

incerto.conformal.prediction_set_from_scores(scores, threshold, descending=True)[source]#

Convert conformity scores to prediction sets.

Parameters:
  • scores (Tensor) – Score tensor of shape (n_samples, n_classes).

  • threshold (float) – Conformity threshold.

  • descending (bool) – If True, higher scores are better.

Return type:

list[Tensor]

Returns:

List of prediction sets (class indices) for each sample.