Skip to content

Commit 872e5e3

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent dd212c4 commit 872e5e3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

machine_learning/gaussian_mixture_model.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,9 @@ def _m_step(self, X, responsibilities):
8585

8686
for k in range(self.n_components):
8787
diff = X - self.means_[k]
88-
self.covariances_[k] = (responsibilities[:, k][:, np.newaxis] * diff).T @ diff
88+
self.covariances_[k] = (
89+
responsibilities[:, k][:, np.newaxis] * diff
90+
).T @ diff
8991
self.covariances_[k] /= Nk[k]
9092
# Add small regularization term for numerical stability
9193
self.covariances_[k] += np.eye(n_features) * 1e-6
@@ -155,4 +157,4 @@ def plot_results(self, X):
155157
gmm = GaussianMixture(n_components=3, max_iter=100, seed=42)
156158
gmm.fit(X)
157159
labels = gmm.predict(X)
158-
gmm.plot_results(X)
160+
gmm.plot_results(X)

0 commit comments

Comments
 (0)