incerto.calibration.logits_to_probs

incerto.calibration.logits_to_probs#

incerto.calibration.logits_to_probs(logits)[source]#

Convert logits to probabilities using softmax.

Parameters:

logits (Tensor) – Tensor of shape (n_samples, n_classes).

Return type:

Tensor

Returns:

Probabilities tensor of same shape.