def test_dirichlet_entropy(self):
alpha = np.array([23, 4, 5, 6, 7])
dirichlet_dist = sp.stats.dirichlet(alpha)
self.assertAlmostEqual\
(dirichlet_dist.entropy(), ef.dirichlet_entropy(alpha))
alpha_shape = (5, 2)
alpha = 10 * np.random.random(alpha_shape)
ef_entropy = ef.dirichlet_entropy(alpha)
dirichlet_entropy = \
[ sp.stats.dirichlet.entropy(alpha[:, k]) for k in range(2) ]
np_test.assert_array_almost_equal(dirichlet_entropy, ef_entropy)
def entropy(self):
return ef.dirichlet_entropy(self['alpha'].get())
