Skip to main content

Distributions

Available distributions

UsageParametersOutputWikipedia link
Bernoulli(prob)prob is a Reala Boolean: TRUE or FALSEBernoulli distribution
Beta(a, b)a and b are Realsa Real in (0, 1)Beta distribution
Binomial(n, prob)n is an Integer and prob is a Realan Integer in 0, ..., nBinomial distribution
Categorical(probs)probs is a sequence of Realsan Integer in 0, ..., K-1 (category index)Categorical distribution
Dirichlet(alphas)alphas is a sequence of Realsa length-K vector on the simplex (non-negative reals summing to 1)Dirichlet distribution
Exponential(rate)rate is a Reala non-negative RealExponential distribution
Gamma(shape, scale)shape and scale are positive Realsa positive RealGamma distribution
Gaussian(mean, stdev)mean and stdev are Realsa Real (-infinity to infinity)Normal distribution
Multinomial(n, probs)n is an Integer and probs is a sequence of Realsa length-K vector of non-negative Integers summing to nMultinomial distribution
Poisson(rate)rate is Reala non-negative IntegerPoisson distribution
Uniform(a, b)a and b are Realsa Real in [a, b]Continuous uniform distribution