incerto.bayesian.disagreement

Contents

incerto.bayesian.disagreement#

incerto.bayesian.disagreement(predictions, method='variance')[source]#

Compute disagreement score for each sample.

Parameters:
  • predictions (Tensor) – Tensor of shape (num_models, batch_size, num_classes)

  • method (str) – Disagreement method (‘variance’, ‘entropy’)

Return type:

Tensor

Returns:

Disagreement scores of shape (batch_size,)