Skip to content

Commit 753c29b

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

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

machine_learning/k_means_clust.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,9 @@ def plot_heterogeneity(heterogeneity, k):
135135
def plot_kmeans(data, centroids, cluster_assignment):
136136
ax = plt.axes(projection="3d")
137137
ax.scatter(data[:, 0], data[:, 1], data[:, 2], c=cluster_assignment, cmap="viridis")
138-
ax.scatter(centroids[:, 0], centroids[:, 1], centroids[:, 2], c="red", s=100, marker="x")
138+
ax.scatter(
139+
centroids[:, 0], centroids[:, 1], centroids[:, 2], c="red", s=100, marker="x"
140+
)
139141
ax.set_xlabel("X")
140142
ax.set_ylabel("Y")
141143
ax.set_zlabel("Z")

0 commit comments

Comments
 (0)