arviz.from_numpyro#

arviz.from_numpyro(posterior=None, *, prior=None, posterior_predictive=None, predictions=None, constant_data=None, predictions_constant_data=None, log_likelihood=None, index_origin=None, coords=None, dims=None, pred_dims=None, num_chains=1)[source]#

Convert NumPyro data into an InferenceData object.

For a usage example read the Creating InferenceData section on from_numpyro

Parameters:
posteriornumpyro.mcmc.MCMC

Fitted MCMC object from NumPyro

prior: dict

Prior samples from a NumPyro model

posterior_predictivedict

Posterior predictive samples for the posterior

predictions: dict

Out of sample predictions

constant_data: dict

Dictionary containing constant data variables mapped to their values.

predictions_constant_data: dict

Constant data used for out-of-sample predictions.

index_originint, optional
coordsdict[str] -> list[str]

Map of dimensions to coordinates

dimsdict[str] -> list[str]

Map variable names to their coordinates

pred_dims: dict

Dims for predictions data. Map variable names to their coordinates.

num_chains: int

Number of chains used for sampling. Ignored if posterior is present.