rbm.py 文件源码

def lnPr(s,p,eps=1e-12,axis=-1):
    '''
    Compute probability of bits s given Bernoulli probabilities p
    Assuming factorized distribution
    \prod p^x (1-p)^(1-x)

    Parameters
    ----------
    s : bits
    p : probability of bits being 1

    Returns
    -------
    '''
    p = p.copy()
    p[p<eps]=eps
    p[p>1-eps]=1-eps
    s = np.int32(s)
    return np.sum(s*slog(p)+(1-s)*np.log1p(-p),axis=axis)