oops

Author: lazyprogrammer

unsupervised_class3/bayes_classifier_gmm.py

@@ -20,8 +20,8 @@ def fit(self, X, Y):
     self.p_y = np.zeros(self.K)
     for k in range(self.K):
       print("Fitting gmm", k)
-      self.p_y[k] = len(Xk)
       Xk = X[Y == k]
+      self.p_y[k] = len(Xk)
       gmm = BayesianGaussianMixture(10)
       gmm.fit(Xk)
       self.gaussians.append(gmm)