Confusion matrix analysis in TensorFlow - Model Pipeline Trace

Model Pipeline - Confusion matrix analysis

This pipeline shows how a model learns to classify data and how we use a confusion matrix to understand its mistakes and successes.

Data Flow - 6 Stages

1Data in

1000 rows x 10 columns→Raw dataset with features and labels→1000 rows x 10 columns

Features: [5.1, 3.5, ...], Label: 0 (class A)

↓

2Preprocessing

1000 rows x 10 columns→Normalize features, encode labels→1000 rows x 10 columns

Normalized features: [0.52, 0.35, ...], Label encoded: 0

↓

3Train/test split

1000 rows x 10 columns→Split data into training and testing sets→Train: 800 rows x 10 columns, Test: 200 rows x 10 columns

Train sample: features [0.52, 0.35, ...], label 0; Test sample: features [0.48, 0.33, ...], label 1

↓

4Model trains

800 rows x 10 columns→Train neural network classifier→Trained model

Model weights updated after each epoch

↓

5Model predicts

200 rows x 10 columns→Model outputs predicted class probabilities→200 rows x 3 classes

Sample prediction: [0.7, 0.2, 0.1]

↓

6Confusion matrix computed

200 true labels, 200 predicted labels→Compare true vs predicted labels to build confusion matrix→3 x 3 matrix

[[50, 2, 3], [4, 45, 1], [2, 3, 90]]

Training Trace - Epoch by Epoch

Loss
1.2 |*****
0.9 |****
0.7 |***
0.5 |**
0.4 |*
    +---------
    Epochs 1-5

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.2	0.45	Model starts learning, accuracy low
2	0.9	0.60	Loss decreases, accuracy improves
3	0.7	0.72	Model learns important patterns
4	0.5	0.80	Good improvement, model stabilizing
5	0.4	0.85	Training converging, accuracy high

Prediction Trace - 4 Layers

Layer 1: Input layer

Layer 2: Hidden layer (ReLU activation)

Layer 3: Output layer (Softmax)

Layer 4: Prediction

Model Quiz - 3 Questions

What does a confusion matrix help us understand?

A How many predictions were correct or wrong for each class

B The speed of model training

C The size of the dataset

D The number of layers in the model

Key Insight

The confusion matrix is a powerful tool to see exactly where a model makes mistakes by comparing true and predicted classes. Watching loss decrease and accuracy increase during training shows the model is learning well.

Practice

1. What does a confusion matrix primarily show in a classification problem?

easy

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

B. The training time of the model

C. The number of features used in the model

D. The learning rate of the optimizer

Solution

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

   A confusion matrix displays how many predictions were correct or wrong for each class in classification.

2. Step 2: Compare options with the definition

   Only The number of correct and incorrect predictions for each class correctly describes this purpose; others relate to unrelated model aspects.

3. Final Answer:

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

4. Quick Check:

   Confusion matrix = correct/wrong counts [OK]

Hint: Confusion matrix counts correct vs wrong predictions per class [OK]

Common Mistakes:

• Confusing confusion matrix with training metrics like loss
• Thinking it shows model parameters
• Assuming it shows training time

2. Which TensorFlow function is used to compute a confusion matrix from true labels and predicted labels?

easy

A. tf.data.Dataset.from_tensor_slices()

B. tf.keras.metrics.Accuracy()

C. tf.math.confusion_matrix()

D. tf.nn.softmax()

Solution

1. Step 1: Identify TensorFlow functions related to confusion matrix

   The function tf.math.confusion_matrix() directly computes the confusion matrix from labels.

2. Step 2: Eliminate unrelated functions

   tf.keras.metrics.Accuracy() measures accuracy, tf.data.Dataset.from_tensor_slices() creates datasets, and tf.nn.softmax() applies softmax activation.

3. Final Answer:

   tf.math.confusion_matrix() -> Option C

4. Quick Check:

   Confusion matrix function = tf.math.confusion_matrix() [OK]

Hint: Use tf.math.confusion_matrix() for confusion matrix [OK]

Common Mistakes:

• Using accuracy metric instead of confusion matrix function
• Confusing dataset creation with confusion matrix
• Mixing activation functions with metrics

3. Given the following code, what is the output confusion matrix?

import tensorflow as tf
true_labels = [0, 1, 2, 2, 0]
pred_labels = [0, 2, 2, 2, 0]
cm = tf.math.confusion_matrix(true_labels, pred_labels)
print(cm.numpy())

medium

A. [[2 0 0]
[0 0 0]
[0 1 2]]

B. [[2 0 0]
[0 1 0]
[0 0 2]]

C. [[1 0 1]
[0 0 1]
[0 0 2]]

D. [[2 0 0]
[0 0 1]
[0 0 2]]

Solution

1. Step 1: Count true vs predicted labels for each class

   Class 0 true: predicted 0 twice; class 1 true: predicted 2 once; class 2 true: predicted 2 twice.

2. Step 2: Build confusion matrix rows

   Row 0: [2 correct (0 predicted as 0), 0 predicted as 1, 0 predicted as 2]; Row 1: [0, 0, 1]; Row 2: [0, 0, 2].

3. Final Answer:

   [[2 0 0] [0 0 1] [0 0 2]] -> Option D

4. Quick Check:

   Count matches matrix entries [OK]

Hint: Count true vs predicted labels per class to fill matrix [OK]

Common Mistakes:

• Mixing up rows and columns in the matrix
• Counting predicted labels instead of true labels
• Forgetting to count all instances

4. Identify the error in this TensorFlow code snippet for confusion matrix calculation:

import tensorflow as tf
true = [0, 1, 1, 0]
pred = [0, 1, 0]
cm = tf.math.confusion_matrix(true, pred)
print(cm)

medium

A. The confusion matrix function is misspelled

B. The true and predicted label lists have different lengths

C. TensorFlow is not imported correctly

D. The print statement is missing parentheses

Solution

1. Step 1: Check input list lengths

   True labels list has 4 elements; predicted labels list has 3 elements, causing mismatch.

2. Step 2: Understand function requirements

   tf.math.confusion_matrix() requires true and predicted labels to be same length to compare element-wise.

3. Final Answer:

   The true and predicted label lists have different lengths -> Option B

4. Quick Check:

   Input lengths must match [OK]

Hint: Ensure true and predicted label lists have same length [OK]

Common Mistakes:

• Ignoring length mismatch error
• Assuming function name typo
• Thinking print syntax is wrong in Python 3

5. You have a 3-class classification problem with classes 0, 1, and 2. After training, your confusion matrix is:

[[50  2  3]
 [ 4 45  1]
 [ 5  2 40]]

Which class has the highest precision, and how do you calculate it from the matrix?

hard

A. Class 1; precision = TP / (TP + FP) = 45 / (45 + 4 + 1)

B. Class 2; precision = TP / (TP + FP) = 40 / (3 + 1 + 40)

C. Class 0; precision = TP / (TP + FP) = 50 / (50 + 4 + 5)

D. Class 1; precision = TP / (TP + FP) = 45 / (2 + 45 + 2)

Solution

1. Step 1: Understand precision calculation

   Precision for a class = True Positives (TP) / (TP + False Positives (FP)). TP is diagonal element; FP is sum of column excluding TP.

2. Step 2: Calculate precision for each class

   Class 0: TP=50; FP=sum column 0 excluding TP=4+5=9; precision=50/(50+9)=50/59≈0.847. Class 1: TP=45; FP=2+2=4; precision=45/(45+4)=45/49≈0.918. Class 2: TP=40; FP=3+1=4; precision=40/(40+4)=40/44≈0.909.

3. Step 3: Identify highest precision

   Class 1 has the highest precision (≈0.918).

4. Final Answer:

   Class 1; precision = TP / (TP + FP) = 45 / (45 + 4 + 1) -> Option A

5. Quick Check:

   Precision = TP / (TP + FP) [OK]

Hint: Precision = TP / (TP + FP) using confusion matrix columns [OK]

Common Mistakes:

• Confusing precision with recall
• Using row sums instead of column sums for FP
• Mixing up TP and FP positions