update em_1d_gmm.py

shogun-toolbox#4990
mohitmajithia · May 4, 2020 · c4593eb · c4593eb
1 parent 9b5793b
commit c4593eb
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/examples/undocumented/python/graphical/em_1d_gmm.py b/examples/undocumented/python/graphical/em_1d_gmm.py
@@ -1,7 +1,6 @@
 from pylab import figure,show,connect,hist,plot,legend
 from numpy import array, append, arange, empty, exp
-from shogun import Gaussian, GMM
-from shogun import RealFeatures
+from shogun import sg
 import util
 
 util.set_title('EM for 1d GMM example')
@@ -12,7 +11,7 @@
 min_change=1e-9
 
 #setup the real GMM
-real_gmm=GMM(3)
+real_gmm=sg.GMM(3)
 
 real_gmm.set_nth_mean(array([-2.0]), 0)
 real_gmm.set_nth_mean(array([0.0]), 1)
@@ -29,10 +28,10 @@
 for i in range(199):
     generated=append(generated, array([real_gmm.sample()]), axis=1)
 
-feat_train=RealFeatures(generated)
+feat_train=sg.RealFeatures(generated)
 
 #train GMM using EM
-est_gmm=GMM(3)
+est_gmm=sg.GMM(3)
 est_gmm.train(feat_train)
 est_gmm.train_em(min_cov, max_iter, min_change)