Skip to content

Implementation of k-means algorithm in C (standard C89/C90)

Notifications You must be signed in to change notification settings

KlimentLagrangiewicz/k-means-in-C

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 

Repository files navigation

k-means-in-C

Implementation of k-means clustering algorithm in C (standard C89/C90)

About k-means

The k-means algorithm is an iterative data clustering algorithm developed by Stuart Lloyd of Bell Labs in the 1950s as a technique for pulse-code modulation.
The main idea of the algorithm is that at each iteration, based on the existing partitioning, the cluster centers are recalculated, then the instances are divided into clusters according to which of the new centers turned out to be closer to a specific instance according to a pre-selected metric.

Input data:

  • $X=\mathrm{x}_{i=1,j=1}^{n,m}$ — description of instances, where n is number of instances, m is number of features;
  • $k \in \left \{1, \ldots, n \right \}$ — number of clusters.

Output data:

  • $Y = \left\{ y_i|y_i\in\left\{0,\ldots,k-1\right\}, i = \overline{\left(1,n\right)}\right\}$ — cluster labels.

Advantages of k-means:

  • Low algorithmic complexity;
  • Easy to implement;
  • The possibility for effective parallelization;
  • The presence of many modifications.

Disadvantages of k-means:

  • Sensitivity to initial cluster centers;
  • Algorithm k-means poorly separates closely spaced clusters with a complex structure;
  • The need for preliminary determination of the number of clusters.

Steps of k-means algorithm:

Step 1. Data preparing (autoscaling): $x_{i,j}=\frac{x_{i,j}-\mathrm{E_{X^{j}}}}{\sigma_{X^{j}}}$;
Step 2. Set initial cluster centers: $C = \left\{c_{i}| c_{i} \in ℝ^{m}, i = \overline{\left(1,k\right)} \right\}$;
Step 3. Calculate the initial partition: $y_{i} = \arg\min\limits_{j}\rho\left(x_{i},c_{j} \right)$;
Step 4. Calculate new cluster centers:
$$l_j=\sum_{i=1}^{n}{\left[y_i\equiv j\right]}$$
$$c_j=\frac{1}{l_j} \sum_{i=1}^{n}{\left[y_i\equiv j\right]\cdot x_i}$$
Step 5. Calculate a new split: $y_{i} = \arg\min\limits_{j}\rho\left(x_{i},c_{j} \right)$;
Step 6. Repeat steps 4, 5 until the split changes.
A visualisation of the first four iterations of the algorithm is shown in figure 1.

Figure 1 — Processing the Old Faithful Geyser dataset using the k-means algorithm

Example of usage

Cloning project and changing current directory:

git clone https://github.com/KlimentLagrangiewicz/k-means-in-C
cd k-means-in-C

Building from source (Linux):

make

Building from source (Windows):

make windows

If building was ok, you can find executable file in bin subdirectory.
Run the program:

./bin/k-means-in-C ./datasets/iris/data.txt 150 4 3 ./datasets/iris/new_result.txt ./datasets/iris/res.txt