incerto.bayesian.ensemble_predictions_to_distribution

incerto.bayesian.ensemble_predictions_to_distribution#

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

Convert ensemble predictions to mean and variance.

Parameters:

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

Return type:

Tuple[Tensor, Tensor]

Returns:

Tuple of (mean, variance)