gDirichlet.py 文件源码

def rvs(self, size=1):
        res = np.zeros(shape=(size, self._K))
        for i in range(size):
            alpha = np.array([random.gammavariate(self._a, 1. / self._b) for local in range(self._K)])
            try:
                res[i,:] = np.random.dirichlet(alpha, 1)
            except:
                res[i, :] = res[i-1,:]
            # res[i, :] = np.array([random.gammavariate(local, 1) for local in alpha])

        return res