incerto.bayesian.compute_disagreement

incerto.bayesian.compute_disagreement#

incerto.bayesian.compute_disagreement(predictions)[source]#

Compute disagreement among ensemble members.

Disagreement is measured as the variance of predictions.

Parameters:

predictions (Tensor) – Tensor of shape (num_models, batch_size, num_classes)

Return type:

Tensor

Returns:

Disagreement score of shape (batch_size,)