(data, centroids, cluster_assignment)
| 133 | |
| 134 | |
| 135 | def plot_kmeans(data, centroids, cluster_assignment): |
| 136 | ax = plt.axes(projection="3d") |
| 137 | ax.scatter(data[:, 0], data[:, 1], data[:, 2], c=cluster_assignment, cmap="viridis") |
| 138 | ax.scatter( |
| 139 | centroids[:, 0], centroids[:, 1], centroids[:, 2], c="red", s=100, marker="x" |
| 140 | ) |
| 141 | ax.set_xlabel("X") |
| 142 | ax.set_ylabel("Y") |
| 143 | ax.set_zlabel("Z") |
| 144 | ax.set_title("3D K-Means Clustering Visualization") |
| 145 | plt.show() |
| 146 | |
| 147 | |
| 148 | def kmeans( |