def model(): c = numpyro.sample("c", dist.Gamma(1, 1)) with handlers.collapse(): probs = numpyro.sample("probs", dist.Beta(c, 2)) numpyro.sample("obs", dist.Bernoulli(probs), obs=data)
def model(): c = numpyro.sample("c", dist.Gamma(1, 1)) with handlers.collapse(): probs = numpyro.sample("probs", dist.Beta(c, 2)) with numpyro.plate("plate", len(data)): numpyro.sample("obs", dist.Binomial(10, probs), obs=data)
def model1(): c1 = numpyro.param("c1", 0.5, constraint=dist.constraints.positive) c0 = numpyro.param("c0", 1.5, constraint=dist.constraints.positive) with handlers.collapse(): probs = numpyro.sample("probs", dist.Beta(c1, c0)) numpyro.sample("obs", dist.Binomial(total_count, probs), obs=data)