K-Means clustering in ML Python - Model Pipeline Trace

Model Pipeline - K-Means clustering

K-Means clustering groups data points into clusters based on their similarity. It finds centers (called centroids) and assigns each point to the nearest center, repeating until stable groups form.

Data Flow - 5 Stages

1Input Data

150 rows x 2 columns→Raw data points with two features each→150 rows x 2 columns

[[5.1, 3.5], [4.9, 3.0], [6.7, 3.1]]

↓

2Initialization

150 rows x 2 columns→Randomly select 3 initial centroids→3 rows x 2 columns

[[5.1, 3.5], [6.7, 3.1], [5.0, 3.6]]

↓

3Assign Clusters

150 rows x 2 columns and 3 centroids→Assign each point to nearest centroid→150 rows x 3 columns (one-hot cluster assignment)

[[1,0,0], [1,0,0], [0,1,0]]

↓

4Update Centroids

150 rows x 3 columns (cluster assignments)→Calculate new centroid as mean of assigned points→3 rows x 2 columns

[[5.0, 3.4], [6.6, 3.0], [5.1, 3.5]]

↓

5Repeat Until Convergence

3 rows x 2 columns (centroids)→Repeat assign and update steps until centroids stabilize→3 rows x 2 columns (final centroids)

[[5.0, 3.4], [6.7, 3.1], [5.1, 3.5]]

Training Trace - Epoch by Epoch

Loss
120.5 |*       
 85.3 |**      
 70.1 |***     
 68.9 |****    
 68.7 |*****   
       --------
       Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	120.5	N/A	Initial centroid positions, high total distance
2	85.3	N/A	Centroids updated, total distance decreased
3	70.1	N/A	Clusters more stable, loss continues to drop
4	68.9	N/A	Small improvement, nearing convergence
5	68.7	N/A	Centroids stabilized, algorithm converged

Prediction Trace - 3 Layers

Layer 1: Input Sample

Layer 2: Distance Calculation

Layer 3: Cluster Assignment

Model Quiz - 3 Questions

What does the 'loss' represent in K-Means clustering?

A Accuracy of predictions

B Number of clusters

C Total distance between points and their assigned centroids

D Number of features in data

Key Insight

K-Means clustering groups data by minimizing the total distance between points and their cluster centers. The iterative process improves cluster quality until centroids stabilize, showing how unsupervised learning finds natural groupings without labels.

Practice

1. What does the n_clusters parameter in K-Means control?

easy

A. The initial random seed for cluster centers

B. The maximum number of iterations for the algorithm

C. The number of groups (clusters) to find in the data

D. The distance metric used to measure similarity

Solution

1. Step 1: Understand the role of n_clusters

   The n_clusters parameter tells K-Means how many groups to split the data into.

2. Step 2: Compare with other parameters

   Other parameters control iterations, distance, or randomness, but n_clusters specifically sets group count.

3. Final Answer:

   The number of groups (clusters) to find in the data -> Option C

4. Quick Check:

   n_clusters = number of groups [OK]

Hint: Remember: n_clusters means how many groups you want [OK]

Common Mistakes:

• Confusing n_clusters with max iterations
• Thinking n_clusters sets distance metric
• Assuming n_clusters controls randomness

2. Which of the following is the correct way to import KMeans from scikit-learn in Python?

easy

A. from sklearn import KMeans.cluster

B. import KMeans from sklearn.cluster

C. import sklearn.cluster.KMeans

D. from sklearn.cluster import KMeans

Solution

1. Step 1: Recall Python import syntax

   Python uses 'from module import class' to import specific classes.

2. Step 2: Match correct syntax for KMeans

   The correct import is from sklearn.cluster import KMeans.

3. Final Answer:

   from sklearn.cluster import KMeans -> Option D

4. Quick Check:

   Correct import syntax = from sklearn.cluster import KMeans [OK]

Hint: Use 'from module import class' for importing classes [OK]

Common Mistakes:

• Using wrong import order
• Trying to import with 'import' and 'from' combined incorrectly
• Using dot notation in import statement

3. Given the code below, what will be the output of print(kmeans.labels_)?

from sklearn.cluster import KMeans
import numpy as np

X = np.array([[1, 2], [1, 4], [1, 0], [10, 2], [10, 4], [10, 0]])
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
print(kmeans.labels_)

medium

A. [0 0 0 1 1 1]

B. [1 1 1 0 0 0]

C. [0 1 0 1 0 1]

D. [1 0 1 0 1 0]

Solution

1. Step 1: Understand data grouping

   The first three points are close together near (1, y), the last three near (10, y), so KMeans will group first three as one cluster and last three as another.

2. Step 2: Check label assignment

   KMeans assigns cluster labels starting from 0. The first cluster gets label 0, second cluster label 1. But the cluster near (10, y) is assigned label 0 and near (1, y) label 1 due to initialization and fitting.

3. Final Answer:

   [1 1 1 0 0 0] -> Option B

4. Quick Check:

   Close points share label 0 or 1 [OK]

Hint: Clusters group close points; labels start at 0 [OK]

Common Mistakes:

• Assuming labels start at 1
• Mixing cluster order
• Ignoring spatial grouping

4. What is wrong with the following K-Means code snippet?

from sklearn.cluster import KMeans
X = [[1, 2], [1, 4], [1, 0]]
kmeans = KMeans(n_clusters=4)
kmeans.fit(X)
print(kmeans.labels_)

medium

A. The number of clusters (4) is greater than the number of data points (3)

B. The data X should be a numpy array, not a list

C. The KMeans class requires a random_state parameter

D. The fit method should be called with labels, not data

Solution

1. Step 1: Check cluster count vs data points

   There are only 3 data points but 4 clusters requested, which is invalid because you cannot have more clusters than points.

2. Step 2: Validate other code parts

   Using a list for X is allowed, random_state is optional, and fit expects data, not labels.

3. Final Answer:

   The number of clusters (4) is greater than the number of data points (3) -> Option A

4. Quick Check:

   Clusters ≤ data points [OK]

Hint: Clusters cannot exceed number of data points [OK]

Common Mistakes:

• Thinking data must be numpy array
• Believing random_state is mandatory
• Confusing fit input as labels

5. You have a dataset with 3 natural groups, but you run K-Means with n_clusters=5. What is the likely effect on the clustering result?

hard

A. K-Means will split some natural groups into smaller clusters, causing over-segmentation

B. K-Means will ignore extra clusters and only find 3 groups

C. K-Means will fail to run and throw an error

D. K-Means will merge groups to reduce clusters to 3 automatically

Solution

1. Step 1: Understand cluster count effect

   Setting more clusters than natural groups causes K-Means to split existing groups into smaller parts.

2. Step 2: Check K-Means behavior

   K-Means does not reduce cluster count automatically or throw error if clusters exceed natural groups; it always finds exactly n_clusters.

3. Final Answer:

   K-Means will split some natural groups into smaller clusters, causing over-segmentation -> Option A

4. Quick Check:

   More clusters than groups = splitting groups [OK]

Hint: More clusters than groups splits groups into smaller parts [OK]

Common Mistakes:

• Thinking K-Means auto-adjusts cluster count
• Expecting error when clusters > natural groups
• Assuming extra clusters are ignored