Precision-recall curves in TensorFlow - Model Pipeline Trace

Model Pipeline - Precision-recall curves

This pipeline trains a binary classifier and evaluates it using precision-recall curves. Precision-recall curves help us understand the trade-off between correctly identifying positive cases and avoiding false alarms.

Data Flow - 6 Stages

1Data Loading

1000 rows x 10 columns→Load dataset with features and binary labels→1000 rows x 10 columns

Feature row: [0.5, 1.2, ..., 0.3], Label: 1

↓

2Train/Test Split

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

Train feature row: [0.7, 0.1, ..., 0.4], Train label: 0

↓

3Feature Normalization

800 rows x 10 columns→Scale features to zero mean and unit variance→800 rows x 10 columns

Normalized feature row: [0.1, -0.3, ..., 0.0]

↓

4Model Training

800 rows x 10 columns→Train binary classifier with sigmoid output→Trained model

Model weights updated after each batch

↓

5Prediction on Test Set

200 rows x 10 columns (normalized)→Apply feature normalization to test set using training statistics, then generate predicted probabilities for positive class→200 rows x 1 column

Predicted probability: 0.85

↓

6Precision-Recall Curve Computation

200 rows x 1 column (probabilities) + 200 labels→Calculate precision and recall at different thresholds→Array of precision and recall pairs

Threshold 0.5: Precision=0.8, Recall=0.75

Training Trace - Epoch by Epoch

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

Epoch	Loss ↓	Accuracy ↑	Observation
1	0.65	0.60	Initial training with moderate loss and accuracy
2	0.50	0.72	Loss decreased, accuracy improved
3	0.40	0.80	Model learning well, loss dropping
4	0.35	0.83	Continued improvement in metrics
5	0.30	0.86	Training converging with good accuracy

Prediction Trace - 4 Layers

Layer 1: Input Features

Layer 2: Feature Normalization

Layer 3: Model Prediction (Sigmoid)

Layer 4: Thresholding for Classification

Model Quiz - 3 Questions

What does a point on the precision-recall curve represent?

A Loss and accuracy at a training epoch

B Precision and recall values at a specific threshold

C Input features after normalization

D Model weights after training

Key Insight

Precision-recall curves are useful to evaluate models especially when classes are imbalanced. They show how precision and recall change as the decision threshold varies, helping to choose the best threshold for the task.

Practice

1. What does a precision-recall curve primarily show in model evaluation?

easy

A. The loss value during training

B. The trade-off between precision and recall for different thresholds

C. The accuracy of the model on the test set

D. The number of false negatives only

Solution

1. Step 1: Understand precision and recall

   Precision measures correct positive predictions, recall measures how many actual positives are found.

2. Step 2: Understand what the curve shows

   The precision-recall curve plots precision against recall at different classification thresholds, showing their trade-off.

3. Final Answer:

   The trade-off between precision and recall for different thresholds -> Option B

4. Quick Check:

   Precision-recall curve = trade-off between precision and recall [OK]

Hint: Precision-recall curves show precision vs recall trade-off [OK]

Common Mistakes:

• Confusing precision-recall curve with accuracy plot
• Thinking it shows loss values
• Assuming it only shows false negatives

2. Which TensorFlow function is used to compute precision and recall values for plotting a precision-recall curve?

easy

A. tf.keras.metrics.Precision() and tf.keras.metrics.Recall()

B. tf.keras.losses.MeanSquaredError()

C. tf.data.Dataset.batch()

D. tf.random.uniform()

Solution

1. Step 1: Identify functions related to precision and recall

   TensorFlow provides tf.keras.metrics.Precision() and tf.keras.metrics.Recall() to compute these metrics.

2. Step 2: Exclude unrelated functions

   MeanSquaredError is a loss, Dataset.batch is for data batching, random.uniform generates random numbers.

3. Final Answer:

   tf.keras.metrics.Precision() and tf.keras.metrics.Recall() -> Option A

4. Quick Check:

   Precision and recall functions = tf.keras.metrics.Precision and tf.keras.metrics.Recall [OK]

Hint: Use tf.keras.metrics.Precision and tf.keras.metrics.Recall for these metrics [OK]

Common Mistakes:

• Using loss functions instead of metric functions
• Confusing data pipeline functions with metrics
• Using random functions for metrics

3. Given the following TensorFlow code snippet, what will be the shape of the precision and recall tensors?

import tensorflow as tf
from sklearn.metrics import precision_recall_curve
y_true = tf.constant([0, 1, 1, 0, 1])
y_scores = tf.constant([0.1, 0.4, 0.35, 0.8, 0.7])
precision, recall, thresholds = precision_recall_curve(y_true, y_scores)
print(precision.shape, recall.shape)

medium

A. (1,) (1,)

B. (5,) (5,)

C. (N,) (N,) where N is number of thresholds + 1

D. Scalar values

Solution

1. Step 1: Understand precision_recall_curve output

   The function returns precision and recall arrays for each threshold plus one extra point, so length is number of thresholds + 1.

2. Step 2: Analyze shapes

   Since thresholds vary, precision and recall are vectors, not scalars or fixed length equal to input size.

3. Final Answer:

   (N,) (N,) where N is number of thresholds + 1 -> Option C

4. Quick Check:

   Precision and recall shapes = vectors with length thresholds+1 [OK]

Hint: precision_recall_curve returns arrays, not scalars [OK]

Common Mistakes:

• Assuming output shapes match input size
• Expecting scalar precision and recall
• Confusing thresholds count with input length

4. The following code attempts to plot a precision-recall curve but raises an error. What is the likely cause?

import tensorflow as tf
import matplotlib.pyplot as plt

y_true = tf.constant([0, 1, 1, 0])
y_scores = tf.constant([0.2, 0.8, 0.6, 0.4])
precision, recall, thresholds = tf.metrics.PrecisionRecallCurve()(y_true, y_scores)
plt.plot(precision, recall)
plt.show()

medium

A. Precision and recall are swapped in the plot call

B. plt.plot requires thresholds as x-axis

C. y_true and y_scores must be numpy arrays, not tensors

D. tf.metrics.PrecisionRecallCurve() does not exist

Solution

1. Step 1: Check TensorFlow API correctness

   TensorFlow does not have tf.metrics.PrecisionRecallCurve(); typically use from sklearn.metrics import precision_recall_curve.

2. Step 2: Understand error cause

   Using a non-existent function causes an AttributeError before plotting.

3. Final Answer:

   tf.metrics.PrecisionRecallCurve() does not exist -> Option D

4. Quick Check:

   Correct approach is sklearn.metrics.precision_recall_curve [OK]

Hint: Use sklearn.metrics.precision_recall_curve, not tf.metrics [OK]

Common Mistakes:

• Using non-existent TensorFlow function for PR curve
• Swapping precision and recall axes
• Assuming tensors can't be plotted

5. You have a highly imbalanced dataset with 1% positive class. You want to evaluate your TensorFlow model's ability to detect positives. Which approach is best to use and why? A) Use accuracy metric only B) Use confusion matrix without thresholds C) Use loss value during training D) Use precision-recall curve to evaluate model

hard

A. Use precision-recall curve to evaluate model

B. Use confusion matrix without thresholds

C. Use loss value during training

D. Use accuracy metric only

Solution

1. Step 1: Understand dataset imbalance impact

   With 1% positives, accuracy can be misleading because predicting all negatives gives high accuracy.

2. Step 2: Choose appropriate evaluation metric

   Precision-recall curves focus on positive class detection and show trade-offs, ideal for imbalanced data.

3. Final Answer:

   Use precision-recall curve to evaluate model -> Option A

4. Quick Check:

   Imbalanced data = use precision-recall curve [OK]

Hint: For imbalanced data, prefer precision-recall curves [OK]

Common Mistakes:

• Relying on accuracy for imbalanced data
• Ignoring threshold effects in confusion matrix
• Using loss value as evaluation metric