def test_average(self):
cluster1 = DPCluster(0.5, self.mu1, self.sig)
cluster3 = DPCluster(0.5, self.mu1, self.sig)
cluster5 = DPCluster(0.5, self.mu1, self.sig)
cluster7 = DPCluster(0.5, self.mu1, self.sig)
cluster2 = DPCluster(0.5, self.mu2, self.sig)
cluster4 = DPCluster(0.5, self.mu2, self.sig)
cluster6 = DPCluster(0.5, self.mu2, self.sig)
cluster8 = DPCluster(0.5, self.mu2, self.sig)
mix = DPMixture(
[
cluster1,
cluster2,
cluster3,
cluster4,
cluster5,
cluster6,
cluster7,
cluster8
], niter=4)
avg = mix.average()
assert len(avg.clusters) == 2
assert all(avg.mus[0] == self.mu1)
assert all(avg.mus[1] == self.mu2)
assert all(avg.sigmas[0] == self.sig)
assert all(avg.sigmas[1] == self.sig)
assert avg.pis[0] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
assert avg.pis[1] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
def testAverage(self):
clst1 = DPCluster(0.5, self.mu1, self.sig)
clst3 = DPCluster(0.5, self.mu1, self.sig)
clst5 = DPCluster(0.5, self.mu1, self.sig)
clst7 = DPCluster(0.5, self.mu1, self.sig)
clst2 = DPCluster(0.5, self.mu2, self.sig)
clst4 = DPCluster(0.5, self.mu2, self.sig)
clst6 = DPCluster(0.5, self.mu2, self.sig)
clst8 = DPCluster(0.5, self.mu2, self.sig)
mix = DPMixture([clst1, clst2, clst3, clst4, clst5, clst6, clst7, clst8], niter=4)
avg = mix.average()
assert len(avg.clusters) == 2
assert all(avg.mus[0] == self.mu1)
assert all(avg.mus[1] == self.mu2)
assert all(avg.sigmas[0] == self.sig)
assert all(avg.sigmas[1] == self.sig)
assert avg.pis[0] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
assert avg.pis[1] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
def testAverage(self):
clst1 = DPCluster(0.5, self.mu1, self.sig)
clst3 = DPCluster(0.5, self.mu1, self.sig)
clst5 = DPCluster(0.5, self.mu1, self.sig)
clst7 = DPCluster(0.5, self.mu1, self.sig)
clst2 = DPCluster(0.5, self.mu2, self.sig)
clst4 = DPCluster(0.5, self.mu2, self.sig)
clst6 = DPCluster(0.5, self.mu2, self.sig)
clst8 = DPCluster(0.5, self.mu2, self.sig)
mix = DPMixture(
[clst1, clst2, clst3, clst4, clst5, clst6, clst7, clst8], niter=4)
avg = mix.average()
assert len(avg.clusters) == 2
assert all(avg.mus[0] == self.mu1)
assert all(avg.mus[1] == self.mu2)
assert all(avg.sigmas[0] == self.sig)
assert all(avg.sigmas[1] == self.sig)
assert avg.pis[0] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
assert avg.pis[1] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
def test_average(self):
cluster1 = DPCluster(0.5, self.mu1, self.sig)
cluster3 = DPCluster(0.5, self.mu1, self.sig)
cluster5 = DPCluster(0.5, self.mu1, self.sig)
cluster7 = DPCluster(0.5, self.mu1, self.sig)
cluster2 = DPCluster(0.5, self.mu2, self.sig)
cluster4 = DPCluster(0.5, self.mu2, self.sig)
cluster6 = DPCluster(0.5, self.mu2, self.sig)
cluster8 = DPCluster(0.5, self.mu2, self.sig)
mix = DPMixture([
cluster1, cluster2, cluster3, cluster4, cluster5, cluster6,
cluster7, cluster8
],
niter=4)
avg = mix.average()
assert len(avg.clusters) == 2
assert all(avg.mus[0] == self.mu1)
assert all(avg.mus[1] == self.mu2)
assert all(avg.sigmas[0] == self.sig)
assert all(avg.sigmas[1] == self.sig)
assert avg.pis[0] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
assert avg.pis[1] == 0.5, 'pis should be 0.5 but is %f' % avg.pis()[0]
