(heterogeneity, k)
| 123 | |
| 124 | |
| 125 | def plot_heterogeneity(heterogeneity, k): |
| 126 | plt.figure(figsize=(7, 4)) |
| 127 | plt.plot(heterogeneity, linewidth=4) |
| 128 | plt.xlabel("# Iterations") |
| 129 | plt.ylabel("Heterogeneity") |
| 130 | plt.title(f"Heterogeneity of clustering over time, K={k:d}") |
| 131 | plt.rcParams.update({"font.size": 16}) |
| 132 | plt.show() |
| 133 | |
| 134 | |
| 135 | def plot_kmeans(data, centroids, cluster_assignment): |
no test coverage detected