def clustering_gmm_modular(fm_train=generated, n=2, min_cov=1e-9, max_iter=1000, min_change=1e-9, cov_type=0):
from shogun.Distribution import GMM
from shogun.Features import RealFeatures
from shogun.Library import Math_init_random
Math_init_random(5)
feat_train = RealFeatures(generated)
est_gmm = GMM(n, cov_type)
est_gmm.train(feat_train)
est_gmm.train_em(min_cov, max_iter, min_change)
return est_gmm
def clustering_gmm_modular (fm_train=generated,n=2,min_cov=1e-9,max_iter=1000,min_change=1e-9,cov_type=0): from shogun.Distribution import GMM from shogun.Features import RealFeatures from shogun.Library import Math_init_random Math_init_random(5) feat_train=RealFeatures(generated) est_gmm=GMM(n, cov_type) est_gmm.train(feat_train) est_gmm.train_em(min_cov, max_iter, min_change) return est_gmm
real_gmm.set_nth_mean(array([2.0, -2.0]), 2)
real_gmm.set_nth_cov(array([[1.0, 0.2], [0.2, 0.5]]), 0)
real_gmm.set_nth_cov(array([[0.2, 0.1], [0.1, 0.5]]), 1)
real_gmm.set_nth_cov(array([[0.3, -0.2], [-0.2, 0.8]]), 2)
real_gmm.set_coef(array([0.3, 0.4, 0.3]))
generated = array([real_gmm.sample()])
for i in range(199):
generated = append(generated, array([real_gmm.sample()]), axis=0)
generated = generated.transpose()
feat_train = RealFeatures(generated)
est_smem_gmm = GMM(3, cov_type)
est_smem_gmm.train(feat_train)
est_smem_gmm.set_nth_mean(array([2.0, 0.0]), 0)
est_smem_gmm.set_nth_mean(array([-2.0, -2.0]), 1)
est_smem_gmm.set_nth_mean(array([-3.0, -3.0]), 2)
est_smem_gmm.set_nth_cov(array([[1.0, 0.0], [0.0, 1.0]]), 0)
est_smem_gmm.set_nth_cov(array([[1.0, 0.0], [0.0, 1.0]]), 1)
est_smem_gmm.set_nth_cov(array([[1.0, 0.0], [0.0, 1.0]]), 2)
est_smem_gmm.set_coef(array([0.3333, 0.3333, 0.3334]))
print est_smem_gmm.train_smem(max_iter, max_cand, min_cov, max_em_iter,
min_change)
est_em_gmm = GMM(3, cov_type)
real_gmm.set_nth_mean(array([0.0]), 1)
real_gmm.set_nth_mean(array([2.0]), 2)
real_gmm.set_nth_cov(array([[0.3]]), 0)
real_gmm.set_nth_cov(array([[0.1]]), 1)
real_gmm.set_nth_cov(array([[0.2]]), 2)
real_gmm.set_coef(array([0.3, 0.5, 0.2]))
generated=array([real_gmm.sample()])
for i in range(199):
generated=append(generated, array([real_gmm.sample()]), axis=1)
feat_train=RealFeatures(generated)
est_gmm=GMM(3)
est_gmm.train(feat_train)
est_gmm.train_em(min_cov, max_iter, min_change)
est_mean1=est_gmm.get_nth_mean(0)
est_mean2=est_gmm.get_nth_mean(1)
est_mean3=est_gmm.get_nth_mean(2)
est_cov1=est_gmm.get_nth_cov(0)
est_cov2=est_gmm.get_nth_cov(1)
est_cov3=est_gmm.get_nth_cov(2)
est_coef=est_gmm.get_coef()
print est_mean1
print est_cov1
print est_mean2
print est_cov2
print est_mean3
print est_cov3
real_gmm.set_nth_mean(array([0.0]), 1)
real_gmm.set_nth_mean(array([2.0]), 2)
real_gmm.set_nth_cov(array([[0.3]]), 0)
real_gmm.set_nth_cov(array([[0.1]]), 1)
real_gmm.set_nth_cov(array([[0.2]]), 2)
real_gmm.set_coef(array([0.3, 0.5, 0.2]))
generated=array([real_gmm.sample()])
for i in range(199):
generated=append(generated, array([real_gmm.sample()]), axis=1)
feat_train=RealFeatures(generated)
est_smem_gmm=GMM(3)
est_smem_gmm.train(feat_train)
print est_smem_gmm.train_smem(max_iter, max_cand, min_cov, max_em_iter, min_change)
est_em_gmm=GMM(3)
est_em_gmm.train(feat_train)
print est_em_gmm.train_em(min_cov, max_em_iter, min_change)
min_gen=min(min(generated))
max_gen=max(max(generated))
plot_real=empty(0)
plot_est_smem=empty(0)
plot_est_em=empty(0)
for i in arange(min_gen, max_gen, 0.001):
plot_real=append(plot_real, array([real_gmm.cluster(array([i]))[3]]))
plot_est_smem=append(plot_est_smem, array([est_smem_gmm.cluster(array([i]))[3]]))
plot_est_em=append(plot_est_em, array([est_em_gmm.cluster(array([i]))[3]]))
real_plot=plot(arange(min_gen, max_gen, 0.001), plot_real, "b")
