posterior_contraction#

bayesflow.diagnostics.posterior_contraction(estimates: Mapping[str, ~numpy.ndarray] | ~numpy.ndarray, targets: Mapping[str, ~numpy.ndarray] | ~numpy.ndarray, variable_keys: Sequence[str] = None, variable_names: Sequence[str] = None, test_quantities: dict[str, ~collections.abc.Callable]=None, aggregation: Callable | None = <function median>) → dict[str, any][source]#

Computes the posterior contraction (PC) from prior to posterior for the given samples.

Parameters:
estimatesnp.ndarray of shape (num_datasets, num_draws_post, num_variables)

Posterior samples, comprising num_draws_post random draws from the posterior distribution for each data set from num_datasets.

targetsnp.ndarray of shape (num_datasets, num_variables)

Prior samples, comprising num_datasets ground truths.

variable_keysSequence[str], optional (default = None)

Select keys from the dictionaries provided in estimates and targets. By default, select all keys.

variable_namesSequence[str], optional (default = None)

Optional variable names to show in the output.

test_quantitiesdict or None, optional, default: None

A dict that maps plot titles to functions that compute test quantities based on estimate/target draws.

The dict keys are automatically added to variable_keys and variable_names. Test quantity functions are expected to accept a dict of draws with shape (batch_size, ...) as the first (typically only) positional argument and return an NumPy array of shape (batch_size,). The functions do not have to deal with an additional sample dimension, as appropriate reshaping is done internally.

aggregationcallable or None, optional (default = np.median)

Function to aggregate the PC across draws. Typically np.mean or np.median. If None is provided, the individual values are returned.

Returns:
resultdict

Dictionary containing:

  • “values”float or np.ndarray

    The (optionally aggregated) posterior contraction per variable

  • “metric_name”str

    The name of the metric (“Posterior Contraction”).

  • “variable_names”str

    The (inferred) variable names.

Notes

Posterior contraction measures the reduction in uncertainty from the prior to the posterior. Values close to 1 indicate strong contraction (high reduction in uncertainty), while values close to 0 indicate low contraction.