def realize_hp(hp, tolerance=1e-3): """ Converts betas to a full (approximate) sample from a DP """ if hp.beta0 > 0: hp.beta0 = 0. betas = stick(hp.gamma, tolerance).values() hp.betas = np.array(betas, dtype=np.float)
def test_stick(): gammas = [.1, 1., 5., 10.] for gamma in gammas: for _ in range(5): betas = stick(gamma).values() assert_almost_equal(sum(betas), 1., places=5)