update

lazyprogrammer · Mar 15, 2023 · 92633f0 · 92633f0
1 parent 95bed3c
commit 92633f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/unsupervised_class3/bayes_classifier_gmm.py b/unsupervised_class3/bayes_classifier_gmm.py
@@ -28,7 +28,7 @@ def fit(self, X, Y):
       print("Fitting gmm", k)
       Xk = X[Y == k]
       self.p_y[k] = len(Xk)
-      gmm = BayesianGaussianMixture(10)
+      gmm = BayesianGaussianMixture(n_components=10)
       gmm.fit(Xk)
       self.gaussians.append(gmm)
     # normalize p(y)