clusters coloured by cluster

app.py

@@ -16,8 +16,8 @@ def compute(number_of_samples, number_of_clusters, randomised_state):
 
     fig = plt.figure()
     ax1 = fig.add_subplot(111)
-    ax1.scatter(scaled_features[:,0], scaled_features[:,1], s=8, c='c', marker="o", label='dataset')
-    ax1.scatter(kmeans.cluster_centers_[:,0],kmeans.cluster_centers_[:,1], s=40, c='m', marker="o", label='cluster centers')
+    ax1.scatter(scaled_features[:,0], scaled_features[:,1], c=kmeans.labels_, s=8, marker="o", label='dataset')
+    ax1.scatter(kmeans.cluster_centers_[:,0],kmeans.cluster_centers_[:,1], s=50, c='r', marker="o", label='cluster centers')
     plt.legend(loc='upper left')
     return fig