incerto.bayesian.ensemble_diversity

incerto.bayesian.ensemble_diversity#

incerto.bayesian.ensemble_diversity(predictions, metric='variance')[source]#

Measure diversity among ensemble members.

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

  • metric (str) – Diversity metric (‘variance’, ‘disagreement’, ‘kl’)

Return type:

float

Returns:

Diversity score (higher = more diverse)