pymc3 使用参数数组生成随机变量

Question

在pymc3中，数组形状的随机变量say 3可以生成如下

y = Normal('y', mu, sigma, shape=3, observed=some_data)

现在假设 y 取决于参数数组 mu = [1,2,3] 和 sigma = [4,5,6] 而不是单个值，我将如何指定它？

Answer 1

我想要一种方法来执行以下操作

for i in range(3):
    y[i] = Normal(mu[i], sigma[i], observed=somedata[i])

方法是简单地传入一个值列表。

y=Normal(mu, sigma, observed=somedata)

其中 mu 和 sigma 是列表

pymc3 generate stochastic variables with array of parameters