在PyMC3中使用复杂的可能性

时间:2015-02-11 00:05:04

标签: pymc pymc3

pymc.__version__ = '3.0'  
theano.__version__ = '0.6.0.dev-RELEASE'

我试图使用具有复杂似然函数的PyMC3:

第一个问题:这可能吗?

我尝试使用Thomas Wiecki's post作为指南:

import numpy as np
import theano as th
import pymc as pm
import scipy as sp

# Actual data I'm trying to fit
x = np.array([52.08, 58.44, 60.0, 65.0, 65.10, 66.0, 70.0, 87.5, 110.0, 126.0])
y = np.array([0.522, 0.659, 0.462, 0.720, 0.609, 0.696, 0.667, 0.870, 0.889,  0.919])
yerr = np.array([0.104,  0.071,  0.138,  0.035,  0.102, 0.096,  0.136,  0.031, 0.024, 0.035])

th.config.compute_test_value = 'off'
a = th.tensor.dscalar('a')

with pm.Model() as model:
    # Priors
    alpha = pm.Normal('alpha', mu=0.3, sd=5)
    sig_alpha = pm.Normal('sig_alpha', mu=0.03, sd=5)
    t_double = pm.Normal('t_double', mu=4, sd=20)
    t_delay = pm.Normal('t_delay', mu=21, sd=20)
    nu = pm.Uniform('nu', lower=0, upper=20)

    # Some functions needed for calculation of the y estimator
    def T(eqd):
        doses = np.array([52.08, 58.44, 60.0, 65.0, 65.10, 
                          66.0, 70.0, 87.5, 110.0, 126.0])
        tmt_times = np.array([29,29,43,29,36,48,22,11,7,8])
        return np.interp(eqd, doses, tmt_times)

    def TCP(a):
        time = T(x)
        BCP = pm.exp(-1E7*pm.exp(-alpha*x*1.2 + 0.69315/t_delay(time-t_double)))
        return pm.prod(BCP)

    def normpdf(a, alpha, sig_alpha):
        return 1./(sig_alpha*pm.sqrt(2.*np.pi))*pm.exp(-pm.sqr(a-alpha)/(2*pm.sqr(sig_alpha)))

    def normcdf(a, alpha, sig_alpha):
        return 1./2.*(1+pm.erf((a-alpha)/(sig_alpha*pm.sqrt(2))))

    def integrand(a):
        return normpdf(a,alpha,sig_alpha)/(1.-normcdf(0,alpha,sig_alpha))*TCP(a)

    func = th.function([a,alpha,sig_alpha,t_double,t_delay], integrand(a))

    y_est = sp.integrate.quad(func(a, alpha, sig_alpha,t_double,t_delay), 0, np.inf)[0]

    likelihood = pm.T('TCP', mu=y_est, nu=nu, observed=y_tcp)

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

产生关于y_est的表达式的以下消息:

  

TypeError :(' theano函数的错误输入参数,名称为":42"索引0(从0开始)','预期类似数组的对象,但找到了一个变量:也许你试图在(可能是共享的)变量而不是数字数组上调用一个函数?')

我已经克服了其他各种障碍来实现这一目标,这就是我被困的地方。所以,如果第一个问题的答案是“是”,那么我是否走在正确的轨道上?任何指导都会有所帮助!

N.B。这是similar question I foundanother

免责声明:我对此非常陌生。我之前唯一的经验是成功地再现了托马斯的线性回归实例。帖子。我也成功运行了Theano测试套件,所以我知道它有效。

1 个答案:

答案 0 :(得分:3)

是的,它有可能制造具有复杂或任意可能性的东西。虽然这看起来不像你在这里做的。看起来您将一个变量复杂转换为另一个变量,即集成步骤。

您的特殊例外是integrate.quad需要一个numpy数组,而不是pymc Variable。如果你想在pymc中做四元组,你必须make a custom theano Op(带衍生物)。