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]