(heterogeneity, k)
| 113 | |
| 114 | from matplotlib import pyplot as plt |
| 115 | def plot_heterogeneity(heterogeneity, k): |
| 116 | plt.figure(figsize=(7,4)) |
| 117 | plt.plot(heterogeneity, linewidth=4) |
| 118 | plt.xlabel('# Iterations') |
| 119 | plt.ylabel('Heterogeneity') |
| 120 | plt.title('Heterogeneity of clustering over time, K={0:d}'.format(k)) |
| 121 | plt.rcParams.update({'font.size': 16}) |
| 122 | plt.show() |
| 123 | |
| 124 | def kmeans(data, k, initial_centroids, maxiter=500, record_heterogeneity=None, verbose=False): |
| 125 | '''This function runs k-means on given data and initial set of centroids. |
no test coverage detected