示例#1
0
文件: tm.py 项目: sbos/twtm
    def theta_t(th, n, p):
        pt = pt_t(th, n, p)
        if binomial(1, pt) == 1:
            return (th, pt, np.log(pt))

        tt = dirichlet(alpha + n, 1)[0]
        return (tt, pt, np.log(1-pt) + dir_logpdf(tt, alpha + n))
示例#2
0
文件: tm.py 项目: sbos/twtm
 def prior_prob(theta):
     return dir_logpdf(theta, alpha)
示例#3
0
文件: tm.py 项目: sbos/twtm
 def trans_prob(curr, prev):
     if np.power(curr - prev, 2).sum() < 1e-10:
         return np.log(p0)
     return np.log(1-p0) + dir_logpdf(curr, alpha)