def test_cluster_num(): for i in [2,3,4,5,6]: Data = np.load("./Data/cluster_"+str(i)+".npy") gmm = GMM(data = Data) gmm.aic() gmm.bic() vbem = VBEM(data = Data) vbem.train() vbem.show()
def test_dimension(): for i in [3,4,5,6,7]: Data = np.load("./Data/dimension_"+str(i)+".npy") gmm = GMM(data = Data) gmm.aic() gmm.bic() vbem = VBEM(data = Data) vbem.train() vbem.show()
def test_sample_sizes(): for i in [10,20,50,100,200,300]: Data = np.load("./Data/data_"+str(i)+".npy") gmm = GMM(data = Data) gmm.aic() gmm.bic() vbem = VBEM(data = Data) vbem.train() vbem.show()