I want to make posterior predictions using Predictive but I don’t want to add the posterior sigma noise to predictions. Would the below be a good way to achieve this in my model function?
def model(x=None, y=None):
...
mu = ...
if y:
pred = numpyro.sample(..., dist.Normal(mu, sigma), obs=y)
else:
return mu # E(y|x)