bug fix for lambda_mats initialization

Author: yuta-nakahara

bayesml/gaussianmixture/_gaussianmixture.py

@@ -67,7 +67,7 @@ def __init__(
         # params
         self.pi_vec = np.ones(self.c_num_classes) / self.c_num_classes
         self.mu_vecs = np.zeros([self.c_num_classes,self.c_degree])
-        self.lambda_mats = np.tile(np.identity(self.c_degree),[self.c_num_classes,self.c_degree,self.c_degree])
+        self.lambda_mats = np.tile(np.identity(self.c_degree),[self.c_num_classes,1,1])
 
         # h_params
         self.h_alpha_vec = np.ones(self.c_num_classes) / 2        

bayesml/gaussianmixture/test.py

@@ -4,13 +4,8 @@
 from time import time
 
 gen_model = gaussianmixture.GenModel(
-    num_classes=2,
-    degree=2,
-    mu_vecs=np.array([[-2,-2],[2,2]]),
+    c_num_classes=2,
+    c_degree=2,
+    pi_vec=np.ones(3) / 3,
     )
-x,z = gen_model.gen_sample(sample_size=100)
-print(x.shape)
-
-learn_model = gaussianmixture.LearnModel(num_classes=10, degree=2)
-learn_model.update_posterior(x)
-learn_model.visualize_posterior()
+print(gen_model.get_params())