コード例 #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)