[scikit-learn] Drawing contours in KMeans

[Mahmood Naderan]

Hi
I use the following code to highlight the cluster centers with some red
dots.

kmeans = KMeans(n_clusters=6, init='k-means++', max_iter=100, n_init=10,
random_state=0)
pred_y = kmeans.fit_predict(a)
plt.scatter(a[:,0], a[:,1])
plt.scatter(kmeans.cluster_centers_[:, 0], kmeans.cluster_centers_[:, 1],
s=100, c='red')
plt.show()

I would like to know if it is possible to draw contours over the clusters.
Is there any way for that?
Please let me know if there is a function or option in KMeans.

Regards,
Mahmood
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mail.python.org/pipermail/scikit-learn/attachments/20201209/78482f46/attachment.html>