python - pymc3 生成带有参数数组的随机变量

Question

在 pymc3 中，可以按如下方式生成数组形状为 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 是列表