Share on facebook
Share on twitter
Share on linkedin
Share on pinterest

Optimal k in K-means

A major challenge in the K-means algorithm is choosing the optimal value of k; however, selecting the right value of k is quite tricky and is also crucial as it can impact the performance of the model.

However, If you don’t know how many clusters you have in advance, then how do you select the ideal value of k?

There are several methods to determine the optimal k in K-means.

We’ll discuss various supervised and unsupervised methods to determine the right value for k.


These methods can be used when we have external information about the data, i.e., if we know the actual ground truth.


It estimates how many of the clusters predicted contain only members of a single class.

Homogeneity is bounded between 0 and 1, higher the score better it is.

It is determined by,

h=1-\frac{H(C | K)}{H(C)}

H(C|K) is the conditional entropy, which measures the uncertainty in determining the right class after having the clustered dataset.

Where C is the number of classes, and K is the number of clusters.

We’ll normalize the value with the entropy of the class H(C) to have a homogeneity score.


Completeness measures whether all members of a given class are assigned to the same cluster.

It is bounded between 0 and 1.

Completeness is symmetrical to homogeneity.

Completeness is determined using the conditional entropy H(C|K), which measures the uncertainty in determining the right cluster given the information of the class.

c=1-\frac{H(K | C)}{H(K)}

The value is normalized with the entropy of the clusters H(K).


V-measure is the harmonic mean of homogeneity and completeness.

v=2 \cdot \frac{h \cdot c}{h+c}

By using this value, we can monitor both the homogeneity and completeness.

Now let’s see how we can use these three metrics to find the optimal number of k.

First, We’ll create a dataset with 3 clusters using the sklearn’s make_blobs.

optimal k in k-means

Now, we can compute the homogeneity, completeness, and v-measure using sklearn for different values of k.

Let’s plot the scores against different values of k.


As you can see in the preceding plot all of the three metrics peaks at k=3. That is the ideal value of k.

sklearn also provides a way to calculate all of these three scores at once using the method homogeneity_completeness_v_measure(y_true, y_pred)



The Rand index is a simple metric based on the agreements and disagreements between pairs of data points in the clusterings.

It measures how similar the induced clusters are to the actual ground truth.

It is defined as,

R=\frac{a+b}{\left(\begin{array}{c}{M} \\ {2}\end{array}\right)}

a represents the number of pair of observations belonging to the same cluster across two different clusterings

b represents the number of pair of observations belonging to different cluster across two different clusterings

m is the number of observations.

Rand index ranges from 0 to 1. Where 1 indicates that the clusterings are exactly the same, and 0 indicates that the clusterings are maximally dissimilar.

However, the complaint with the Rand index is that the expected value of the Rand index between two random partitions is not a constant(say zero), i.e., even if we cluster points randomly, we will get a value greater than 0.


To counter the problems faced by Rand Index, Hubert and Arabie suggest an Adjusted Rand index.

ARI has zero expected value in the case of random partition.

It is defined as,

A R I=\frac{\text { Index }-\text { Expected Index }}{\text { Maximum Index -Expected Index }}

This ARI is bounded between -1 and 1. Where 1 indicates that the clusters are identical.

optimal k in K-means

Visit this link to learn more about the Rand index and this link to learn more about the Adjusted Rand Index.

We can use all the above-discussed metrics only if we have prior information about the data or if we have the actual ground truth.

In clustering, this is rarely the case, as clustering in most of the time is applied as unsupervised.


These methods are based on the intrinsic parameters of the clustering algorithm and don’t need any external information about the data.


The first method we are going to see in this section is the elbow method.

The elbow method plots the value of inertia produced by different values of k.

The value of inertia will decline as k increases.

The idea here is to choose the value of k after which the inertia doesn’t decrease significantly anymore.

optimal k in K-means

If you eyeball the preceding diagram, you can see an elbow like inflection at k=3.

That is the value of k we have to choose.


Another unsupervised metric we are going to discuss is the silhouette coefficient.

The silhouette coefficient combines the idea of cluster cohesion and cluster separation.

Let’s take a step back and understand what cohesion and separation are

COHESION: It measures how similar observation is to the assigned cluster.

It is measured by the within-cluster sum of squares(WSS).

WSS=\sum \sum_{x \in C_{i}}\left(x-c_{i}\right)^{2}

SEPARATION: It measures how dissimilar an observation is to the observation of nearby cluster. In simple words, it measures how well a cluster is separated from other clusters.

It is measured by between cluster sum of squares(BSS).


Ci is the number of observations in each group.

The silhouette coefficient is determined by,

S C=\frac{b-a}{\max (a, b)}

a: mean distance to all other points in its cluster

b: mean distance to all other points in the next nearest cluster

The silhouette score ranges from -1 to 1.  The better it is if the score is near to 1.

optimal k in K-means


The Davies-Bouldin index is based on the ratio between compactness to separation.

The DB-index is calculated using the formula,

D B=\frac{1}{n} \sum_{i=1}^{n} \max _{j \neq i}\left(\frac{\sigma_{i}+\sigma_{j}}{d\left(c_{i}, c_{j}\right)}\right)

Where n is the number of clusters, ci is the centroid of cluster i, σi is the average distance of all observations in cluster i, and d(ci,cj) is the distance between clusters i and j.

It is bounded between 0 to 1, lower is better.


The optimum number of cluster corresponds to the minimum value of DB-index. In our case, it is 3.


The CH-index is another metric which can be used to find the best value of k using with-cluster-sum-of-squares(WSS) and between-cluster-sum-of-squares(BSS).

WSS measures the compactness of the cluster, while the BSS is a measure of separation.

CH-index is defined as,

\mathrm{CH}(K)=\frac{B(K) /(K-1)}{W(K) /(n-K)}

Where B is between-sum-of-squares, W is within-sum-of-square, n is the number of data points and k is the number of clusters.

A higher score indicates that the clusters are well separated.



The idea behind the gap statistic is to compare the WSS value for different values of k on our actual dataset versus a null reference distribution of the data.

The reference distribution is usually a random uniform distribution.

To find the ideal number of clusters, we choose a value of k that gives us the maximum value of the Gap statistic.

The gap statistic for a given k is defined as follows,

\operatorname{Gap}(k)=E\left(\log \left(W_{k}\right)\right)-\log \left(W_{k}\right)

Where E\left(\log \left(W_{k}\right)\right) indicates the expected value of \log \left(W_{k}\right) under the null distribution.

Unfortunately, sklearn doesn’t have an implementation of gap statistic. We need to install an additional package gap_stat.

You may either use this link or this to install the package.

gap statistic


In this post, we have discussed various methods to find the optimal value of k in the K-means algorithm.

I hope this article gives some insight on choosing the ideal value of k.

If you want to learn more metrics, Milligan and Cooper have presented a comparison study over thirty validity indexes. You can read it here.

Complete code for this tutorial can be found in this Github Repo.

Love What you Read. Subscribe to our Newsletter.

Stay up to date! We’ll send the content straight to your inbox, once a week. We promise not to spam you.

Leave a Reply

Your email address will not be published. Required fields are marked *

Subscribe Now! We'll keep you updated.