PyMC3中的正弦回归

2024-10-03 02:38:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我通过一个回归示例来研究PyMC3。我从一条直线开始,然后转到二次曲线,效果很好。当我试图转移到一个带有随机变量的正弦函数时,事情变糟了。在

这是我的PyMC3代码:

import pymc as pm
import numpy as np

trace = None
with pm.Model() as model:
    alpha = pm.Normal('alpha', mu=0, sd=20)
    beta = pm.Normal('beta', mu=0, sd=20)
    mewmew = pm.Normal('mewmew', mu=0, sd=20)
    sigma = pm.Uniform('sigma', lower=0, upper=20)

    y_est = alpha + beta * numpy.sin(mewmew * x)

    likelihood = pm.Normal('y', mu=y_est, sd=sigma, observed=y)

    start = pm.find_MAP()
    step = pm.NUTS(state=start)
    trace = pm.sample(2000, step, start=start, progressbar=False)

    pm.traceplot(trace);

我得到这个错误:

^{pr2}$

提前谢谢!在


Tags: importalphanumpyastracesdstartsigma