arviz.from_pystan#
- arviz.from_pystan(posterior=None, *, posterior_predictive=None, predictions=None, prior=None, prior_predictive=None, observed_data=None, constant_data=None, predictions_constant_data=None, log_likelihood=None, coords=None, dims=None, posterior_model=None, prior_model=None, save_warmup=None, dtypes=None)[source]#
Convert PyStan data into an InferenceData object.
For a usage example read the Creating InferenceData section on from_pystan
- Parameters
- posteriorStanFit4Model or stan.fit.Fit
PyStan fit object for posterior.
- posterior_predictivestr, a list of str
Posterior predictive samples for the posterior.
- predictionsstr, a list of str
Out-of-sample predictions for the posterior.
- priorStanFit4Model or stan.fit.Fit
PyStan fit object for prior.
- prior_predictivestr, a list of str
Posterior predictive samples for the prior.
- observed_datastr or a list of str
observed data used in the sampling. Observed data is extracted from the
posterior.data
. PyStan3 needs model object for the extraction. Seeposterior_model
.- constant_datastr or list of str
Constants relevant to the model (i.e. x values in a linear regression).
- predictions_constant_datastr or list of str
Constants relevant to the model predictions (i.e. new x values in a linear regression).
- log_likelihooddict of {str: str}, list of str or str, optional
Pointwise log_likelihood for the data. log_likelihood is extracted from the posterior. It is recommended to use this argument as a dictionary whose keys are observed variable names and its values are the variables storing log likelihood arrays in the Stan code. In other cases, a dictionary with keys equal to its values is used. By default, if a variable
log_lik
is present in the Stan model, it will be retrieved as pointwise log likelihood values. UseFalse
or setdata.log_likelihood
to false to avoid this behaviour.- coordsdict[str, iterable]
A dictionary containing the values that are used as index. The key is the name of the dimension, the values are the index values.
- dimsdict[str, List(str)]
A mapping from variables to a list of coordinate names for the variable.
- posterior_modelstan.model.Model
PyStan3 specific model object. Needed for automatic dtype parsing and for the extraction of observed data.
- prior_modelstan.model.Model
PyStan3 specific model object. Needed for automatic dtype parsing.
- save_warmupbool
Save warmup iterations into InferenceData object. If not defined, use default defined by the rcParams.
- dtypes: dict
A dictionary containing dtype information (int, float) for parameters. By default dtype information is extracted from the model code. Model code is extracted from fit object in PyStan 2 and from model object in PyStan 3.
- Returns
- InferenceData object