# 1d EMG Priors
mu_emg = pm.Normal('mu_emg',mu = 0.,sd = 5.,
shape=N_dp)
sigma_emg = pm.HalfCauchy('sigma_emg',beta = 1.,shape = N_dp)
tau_emg = pm.Uniform("tau_emg",lower = .1,upper = 10, shape=N_dp)
lamb_emg = pm.Deterministic("lamb_emg",1./tau_emg)
# Observations
points = pm.DensityDist("obs",lambda value:
logp_emg(value[:,0],mu_emg[category],sigma_emg[category],
lamb_emg[category])+\
mvNormal_logp(mu_norm[category],tau_norm[category],value[:,1:]),
observed = data)
############################
# Example of Sort Run Code #
############################
# n_step = 25000
# start = time.time()
# with model:
# step0 = pm.Metropolis(vars=[alpha,n_lkj])
# step1 = pm.Metropolis(vars=[mu_norm,sigma_norm,lkj_norm,
# mu_emg,sigma_emg,tau_emg])
# step2 = pm.Metropolis(vars=[v])
tt.dot(lkj_structure[0],corr_triangle0.reshape((1,-1)))+\
tt.dot(lkj_structure[1],corr_triangle1.reshape((1,-1))) )
# corr_matrix = corr_triangle[tri_index]
# corr_matrix = tt.fill_diagonal(corr_matrix, 1)
# cov_matrix = tt.diag(sigma).dot(corr_matrix.dot(tt.diag(sigma)))
cov = pm.Deterministic("cov",cov_funct_special(lkj,sigma,tri_index))
tau = pm.Deterministic("tau",tau_funct(cov))
points = pm.DensityDist("obs",lambda value:
mvNormal_logp(mu[category],tau[category],value),
observed = dataset)
n = 25000
start = time.time()
with model_pre_multiD:
step = pm.Metropolis(vars=[alpha,n_lkj,mu,sigma,corr_triangle0,corr_triangle1])
step1 = pm.Metropolis(vars=[v])
step2 = pm.ElemwiseCategorical(vars=[category],
values = list(np.arange(N_dp)))
tr = pm.sample(n, step=[step,step1,step2])
end = time.time()
tr_storage = dict()
