jax.scipy.stats.betabinom.pmf

jax.scipy.stats.betabinom.pmf(k, n, a, b, loc=0)

Beta-二项分布概率质量函数。

JAX 对 scipy.stats.betabinom pmf 的实现。

Beta-二项分布的概率质量函数定义为

\[f(k, n, a, b) = {n \choose k}\frac{B(k+a,n-k-b)}{B(a,b)}\]

其中 \(B(a, b)\) 是 beta() 函数。它定义在 \(n\ge 0\), \(a>0\), \(b>0\), 和非负整数 k 的情况下。

参数:

• k (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，用于评估 PMF 的值

• n (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布形状参数

• a (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布形状参数

• b (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布形状参数

• loc (Array | ndarray | bool | number | bool | int | float | complex) – 类似数组，分布偏移参数

返回:

pmf 值的数组

返回类型:

Array

参见

jax.scipy.stats.betabinom.logpmf()