In Bayesian Hierarchical Linear Regression — NumPyro documentation I can see that the inference data is imported to arviz using data = az.from_numpyro(mcmc)
.
When displaying that object, I can see the data set observed_data
and it includes the variable obs
.
And I assume this is because of numpyro.sample("obs", dist.Normal(FVC_est, σ), obs=FVC_obs)
My question is how I can store other inputs, like Weeks
in this model.
I really want to include that into the arviz data, so I can make plots directly from that object, using both Weeks
& FVC_obs