如何使用pymc查找初始状态的变量以将最终状态与观察值匹配?

时间:2019-07-05 15:11:45

标签: python-2.7 pymc

我有一个初始状态,可以用高斯函数表示,例如F(x),它带有三个自由参数“ amplitude, centroid, sigma”。 我必须用一个转换函数F(x)和两个自由参数G(x)= exp(a*x)+ba来总结b

我想使用pymc查找这五个自由参数,以便最终状态F(x)+G(x)表示具有以下特征的高斯函数:

amplitude=3.05
centroid=5.45
sigma=5.47

我已经查看了此链接以及其他各种问题和答案:Fit two normal distributions (histograms) with MCMC using pymc? 并且编写了相同的代码,但是我不知道应该在哪里输入前三个值。

import numpy as np
import matplotlib.pyplot as pl
from scipy.optimize import curve_fit
import pymc as mc



def GaussFunc(x, A, mu, sigma):
    return A * np.exp(-0.5 * ((x - mu) / sigma)**2)

def trFunc(x,a,b):
    return np.exp(a*x)+b


interval=np.arange(0, 10, 0.05)
A_ini=2.0
mu_ini=3.0
sigma_ini=1.0
initial=GaussFunc(interval, A_ini, mu_ini, sigma_ini, )



intervalT=np.arange(0, np.pi, np.pi/200.0)
a_t=0.2
b_t=-2.0
transf= trFunc(intervalT,a_t,b_t,)

final=np.zeros(200)
final=initial+transf

est_centroid_one = mc.Uniform("est_centroid_one", 0, 10 )
est_sigma_one = mc.Uniform( "est_sigma_one", 0, 5 )
est_height_one = mc.Uniform( "est_height_one", 0, 5 )

est_a_two = mc.Uniform("est_a_two", 0, 1 )
est_b_two = mc.Uniform("est_b_two", -3, 3 )

precision= 1./mc.Uniform("std", 0, 1)**2



@mc.deterministic( trace = False) 
def est_profile_1(x = interval, mu = est_centroid_one, sigma = est_sigma_one, A= est_height_one):
    return GaussFunc( x, A, mu, sigma )


@mc.deterministic( trace = False) 
def est_profile_2(x = intervalT, a = est_a_two, b = est_b_two):
    return trFunc( x, a, b )

@mc.deterministic( trace = False )
def mean( profile_1 = est_profile_1, profile_2 = est_profile_2 ):
    return profile_1 + profile_2

observations = mc.Normal("obs", mean, precision, value = final, observed = True)

model = mc.Model([est_centroid_one, 
                est_height_one,
                est_sigma_one,
                est_a_two,
                est_b_two,
                precision])

map_ = mc.MAP( model )
map_.fit()

mcmc = mc.MCMC( model )
mcmc.sample( 50000,40000 )


print est_centroid_one,est_height_one,est_sigma_one,est_a_two,est_b_two

谢谢

0 个答案:

没有答案