Confusion matrix in ML Python - Model Pipeline Trace

Model Pipeline - Confusion matrix

A confusion matrix helps us see how well a classification model is doing by showing the counts of correct and wrong predictions for each class.

Data Flow - 3 Stages

1Data in

1000 rows x 5 columns→Dataset with features and true labels→1000 rows x 5 columns

Features: [5.1, 3.5, 1.4, 0.2], Label: 'Setosa'

↓

2Model prediction

1000 rows x 4 columns→Model predicts class labels for each row→1000 rows x 1 column

Predicted labels: ['Setosa', 'Versicolor', 'Setosa', ...]

↓

3Confusion matrix calculation

True labels: 1000 rows x 1 column, Predicted labels: 1000 rows x 1 column→Count matches and mismatches between true and predicted labels→Matrix of size 3 x 3 (for 3 classes)

[[50, 2, 0], [1, 45, 4], [0, 3, 40]]

Training Trace - Epoch by Epoch

Loss
0.7 |****
0.6 |*** 
0.5 |**  
0.4 |**  
0.3 |*   
0.2 |    
     1  2  3 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	0.65	0.70	Model starts learning with moderate accuracy
2	0.45	0.82	Loss decreases and accuracy improves
3	0.30	0.90	Model converges with good accuracy

Prediction Trace - 3 Layers

Layer 1: Input sample

Layer 2: Model prediction

Layer 3: Compare with true label

Model Quiz - 3 Questions

What does a confusion matrix show?

A Counts of correct and wrong predictions for each class

B The model's training loss over time

C The input features of the dataset

D The number of layers in the model

Key Insight

A confusion matrix gives a clear picture of where a classification model gets things right or wrong, helping us understand its strengths and weaknesses beyond just overall accuracy.

Practice

1.

What does a confusion matrix primarily show in machine learning?

easy

A. The number of correct and incorrect predictions for each class

B. The speed of the model training process

C. The size of the dataset used for training

D. The number of features in the dataset

Solution

1. Step 1: Understand the purpose of a confusion matrix

   A confusion matrix shows how many times a model predicted each class correctly or incorrectly.

2. Step 2: Identify what it displays

   It displays counts of true positives, true negatives, false positives, and false negatives for each class.

3. Final Answer:

   The number of correct and incorrect predictions for each class -> Option A

4. Quick Check:

   Confusion matrix = correct and incorrect counts [OK]

Hint: Confusion matrix counts correct and wrong guesses per class [OK]

Common Mistakes:

• Thinking it shows training speed
• Confusing it with dataset size
• Assuming it shows feature count

2.

Which of the following is the correct way to represent a confusion matrix for a binary classifier?

[[TP, FP],
 [FN, TN]]

easy

A. true Positives, false Positives in first row; false Negatives, true Negatives in second row

B. true Negatives, false Negatives in first row; false Positives, true Positives in second row

C. false Positives, true Positives in first row; true Negatives, false Negatives in second row

D. false Negatives, true Negatives in first row; true Positives, false Positives in second row

Solution

1. Step 1: Recall confusion matrix layout for binary classification

   The standard layout is: first row has true Positives (TP) and false Positives (FP), second row has false Negatives (FN) and true Negatives (TN).

2. Step 2: Match the given matrix to the layout

   The given matrix [[TP, FP], [FN, TN]] matches the standard layout exactly.

3. Final Answer:

   true Positives, false Positives in first row; false Negatives, true Negatives in second row -> Option A

4. Quick Check:

   Binary confusion matrix = [[TP, FP], [FN, TN]] [OK]

Hint: Remember TP and FP are in the first row [OK]

Common Mistakes:

• Swapping rows and columns
• Confusing FP with FN positions
• Mixing up TP and TN placement

3.

Given the confusion matrix for a 3-class classifier:

[[5, 2, 0],
 [1, 7, 1],
 [0, 2, 6]]

What is the total number of correct predictions?

medium

A. 22

B. 20

C. 21

D. 18

Solution

1. Step 1: Identify correct predictions in confusion matrix

   Correct predictions are on the diagonal: 5 (class 1), 7 (class 2), and 6 (class 3).

2. Step 2: Sum the diagonal values

   5 + 7 + 6 = 18 correct predictions in total.

3. Final Answer:

   18 -> Option D

4. Quick Check:

   Sum diagonal = 18 correct predictions [OK]

Hint: Sum diagonal elements for correct predictions [OK]

Common Mistakes:

• Summing all matrix elements instead of diagonal
• Adding off-diagonal values by mistake
• Ignoring one of the diagonal values

4.

Consider this confusion matrix for a binary classifier:

[[50, 10],
 [5, 35]]

There is a mistake in interpreting the false Positives (FP). Which is the correct FP count?

medium

A. 50

B. 5

C. 10

D. 35

Solution

1. Step 1: Recall FP position in confusion matrix

   In a binary confusion matrix [[TP, FP], [FN, TN]], FP is at position [0][1].

2. Step 2: Identify FP value in given matrix

   Value at [0][1] is 10, so FP = 10.

3. Final Answer:

   10 -> Option C

4. Quick Check:

   FP = matrix[0][1] = 10 [OK]

Hint: FP is always top-right in binary confusion matrix [OK]

Common Mistakes:

• Confusing FP with FN
• Taking FP as bottom-left value
• Mixing FP with TN or TP

5.

A model's confusion matrix for a 3-class problem is:

[[40, 5, 0],
 [3, 30, 7],
 [0, 2, 38]]

Calculate the overall accuracy of the model.

hard

A. 0.82

B. 0.85

C. 0.88

D. 0.90

Solution

1. Step 1: Calculate total correct predictions

   Sum diagonal: 40 + 30 + 38 = 108 correct predictions.

2. Step 2: Calculate total predictions

   Sum all elements: (40+5+0) + (3+30+7) + (0+2+38) = 45 + 40 + 40 = 125 total predictions.

3. Step 3: Compute accuracy

   Accuracy = correct / total = 108 / 125 = 0.864, rounded to 0.86.

4. Final Answer:

   0.86 -> Option B

5. Quick Check:

   Accuracy = 108/125 ≈ 0.86 [OK]

Hint: Accuracy = sum diagonal / sum all elements [OK]

Common Mistakes:

• Using only one class's correct count
• Forgetting to sum all matrix elements
• Rounding incorrectly