TestHDPTopicOrder.py 文件源码

def L_hdp(beta, omega, Tvec, alpha):
    ''' Compute top-tier of hdp bound.
    '''
    K = omega.size
    assert K == beta.size
    assert K == Tvec.size
    rho = OROB.beta2rho(beta, K)
    eta1 = rho * omega
    eta0 = (1-rho) * omega
    digamma_omega = digamma(omega)
    digamma_eta1 = digamma(eta1)
    digamma_eta0 = digamma(eta0)

    ElogU = digamma_eta1 - digamma_omega
    Elog1mU = digamma_eta0 - digamma_omega
    Lscore = \
        np.sum(gammaln(eta1) + gammaln(eta0) - gammaln(omega)) \
        + np.inner(nDoc + 1 - eta1, ElogU) \
        + np.inner(nDoc * OROB.kvec(K) + gamma - eta0, Elog1mU) \
        + alpha * np.inner(beta, Tvec)
    return Lscore