to_datatree#

jeanspy.sampler_numpyro.NumPyroSampler.to_datatree

Class and construction: NumPyroSampler.

NumPyroSampler.to_datatree(**arviz_kwargs)[source]#

Convert the wrapped MCMC result to ArviZ storage.

Notes

Inputs and units. Keyword arguments forwarded to arviz_converter.

Returns and shape. xarray.DataTree with chain/draw dimensions; the converter must return this format.

Validity. The wrapped MCMC must contain a result supported by the chosen converter.

Errors. TypeError if the converter returns anything other than xarray.DataTree; converter exceptions propagate.

Differentiation. Host-side sampling/storage control; this method has no physical-parameter derivative.