ROC curve and AUC in ML Python - Model Pipeline Trace

Model Pipeline - ROC curve and AUC

This pipeline shows how a binary classifier learns to separate two classes and how we evaluate it using the ROC curve and AUC score. The ROC curve helps us see the trade-off between true positive rate and false positive rate at different thresholds. The AUC summarizes this curve into one number showing overall performance.

Data Flow - 6 Stages

1Raw Data Input

1000 rows x 5 columns→Collect features and binary labels→1000 rows x 5 columns

Features: [age=25, income=50000, ...], Label: 0 or 1

↓

2Train/Test Split

1000 rows x 5 columns→Split data into training and testing sets→800 rows x 5 columns (train), 200 rows x 5 columns (test)

Train: 800 samples with labels, Test: 200 samples with labels

↓

3Model Training

800 rows x 5 columns→Train binary classifier (e.g., logistic regression)→Trained model

Model learns weights for features to predict class 0 or 1

↓

4Prediction on Test Set

200 rows x 5 columns→Model outputs probability scores for class 1→200 rows x 1 column (probabilities)

Sample: [features], Output: 0.85 (probability of class 1)

↓

5ROC Curve Calculation

200 rows x 1 column (probabilities) + 200 labels→Calculate true positive rate and false positive rate at various thresholds→Array of (FPR, TPR) points

Threshold 0.5: TPR=0.8, FPR=0.2; Threshold 0.7: TPR=0.6, FPR=0.1

↓

6AUC Computation

Array of (FPR, TPR) points→Calculate area under the ROC curve→Single scalar value between 0 and 1

AUC = 0.85

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	Model starts learning, loss is high, accuracy is low
2	0.50	0.72	Loss decreases, accuracy improves
3	0.40	0.80	Model is learning well, loss continues to drop
4	0.35	0.83	Loss decreases slower, accuracy stabilizes
5	0.32	0.85	Model converges with good accuracy

Prediction Trace - 5 Layers

Layer 1: Input Features

Layer 2: Model Prediction (Logistic Regression)

Layer 3: Thresholding for Classification

Layer 4: ROC Curve Point Calculation

Layer 5: AUC Calculation

Model Quiz - 3 Questions

What does a point on the ROC curve represent?

A The accuracy of the model on the test set

B A pair of true positive rate and false positive rate at a specific threshold

C The loss value during training

D The predicted class label for one sample

Key Insight

The ROC curve and AUC provide a clear way to evaluate how well a binary classifier can distinguish between classes across all possible thresholds, beyond just accuracy. Watching loss decrease and accuracy increase during training shows the model is learning, and the AUC summarizes its overall classification power.

Practice

1. What does the ROC curve primarily show in a classification model?

easy

A. The trade-off between true positive rate and false positive rate at different thresholds

B. The accuracy of the model on the training data

C. The number of correct predictions made by the model

D. The loss value during model training

Solution

1. Step 1: Understand ROC curve components

   The ROC curve plots the true positive rate (sensitivity) against the false positive rate (1 - specificity) at various threshold settings.

2. Step 2: Identify what ROC shows

   This curve shows how well the model can distinguish between classes by varying the decision threshold.

3. Final Answer:

   The trade-off between true positive rate and false positive rate at different thresholds -> Option A

4. Quick Check:

   ROC curve = True Positive Rate vs False Positive Rate [OK]

Hint: ROC curve plots TPR vs FPR at thresholds [OK]

Common Mistakes:

• Confusing ROC with accuracy
• Thinking ROC shows loss values
• Assuming ROC shows only one threshold

2. Which of the following Python code snippets correctly imports the function to compute AUC from sklearn?

easy

A. import roc_auc_score from sklearn.metrics

B. from sklearn.metrics import auc_roc_score

C. from sklearn.metrics import roc_auc_score

D. import sklearn.metrics.roc_auc_score

Solution

1. Step 1: Recall correct import syntax

   In Python, functions are imported using 'from module import function' syntax.

2. Step 2: Identify correct function name

   The correct function name to compute AUC is 'roc_auc_score' from 'sklearn.metrics'.

3. Final Answer:

   from sklearn.metrics import roc_auc_score -> Option C

4. Quick Check:

   Correct import syntax = from sklearn.metrics import roc_auc_score [OK]

Hint: Use 'from sklearn.metrics import roc_auc_score' [OK]

Common Mistakes:

• Using wrong import syntax
• Misspelling function name
• Trying to import module as function

3. Given the following code, what is the printed AUC value?

from sklearn.metrics import roc_auc_score
true_labels = [0, 0, 1, 1]
predicted_scores = [0.1, 0.4, 0.35, 0.8]
auc = roc_auc_score(true_labels, predicted_scores)
print(round(auc, 2))

medium

A. 0.75

B. 0.85

C. 0.65

D. 0.95

Solution

1. Step 1: Understand roc_auc_score calculation

   The AUC measures how well the predicted scores rank positive samples higher than negative ones.

2. Step 2: Calculate AUC for given data

   Using the pairs, the AUC is 0.75 as the positive samples are ranked correctly most of the time but not perfectly.

3. Final Answer:

   0.75 -> Option A

4. Quick Check:

   AUC for given scores = 0.75 [OK]

Hint: Use roc_auc_score to get AUC directly [OK]

Common Mistakes:

• Confusing AUC with accuracy
• Rounding incorrectly
• Mixing up true labels and scores

4. Identify the error in the following code that tries to plot an ROC curve:

from sklearn.metrics import roc_curve
fpr, tpr, thresholds = roc_curve(y_true, y_pred)
plt.plot(fpr, tpr)
plt.show()

medium

A. roc_curve function is not imported correctly

B. y_pred should be predicted probabilities, not class labels

C. plt.plot should use thresholds instead of fpr

D. Missing import statement for numpy

Solution

1. Step 1: Check input requirements for roc_curve

   The roc_curve function requires true labels and predicted scores or probabilities, not class labels.

2. Step 2: Identify the error in inputs

   Using predicted class labels instead of probabilities causes incorrect ROC curve calculation.

3. Final Answer:

   y_pred should be predicted probabilities, not class labels -> Option B

4. Quick Check:

   Use probabilities for roc_curve inputs [OK]

Hint: Use predicted probabilities, not labels, for roc_curve [OK]

Common Mistakes:

• Passing class labels instead of probabilities
• Forgetting to import matplotlib.pyplot as plt
• Plotting thresholds instead of FPR and TPR

5. You have two models with these AUC scores: Model A = 0.82, Model B = 0.79. You want to improve Model B's performance. Which approach is best to increase the AUC score?

hard

A. Reduce the size of the training dataset to avoid overfitting

B. Tune the classification threshold to balance true positive and false positive rates

C. Use accuracy as the main metric instead of AUC

D. Increase the number of training epochs without changing data or model

Solution

1. Step 1: Understand what affects AUC

   AUC measures the model's ranking ability and is threshold-independent.

2. Step 2: Choose the best method to improve AUC

   Tuning the classification threshold does not improve AUC because AUC is threshold-independent; instead, improving the model's discriminative ability or training is needed.

3. Step 3: Evaluate options

   Increasing training epochs (Increase the number of training epochs without changing data or model) can help but may cause overfitting if not managed properly. Reducing dataset size (Reduce the size of the training dataset to avoid overfitting) usually harms performance. Using accuracy (Use accuracy as the main metric instead of AUC) does not improve AUC. Tuning threshold (Tune the classification threshold to balance true positive and false positive rates) affects operating point but not AUC.

4. Final Answer:

   Increase the number of training epochs without changing data or model -> Option D

5. Quick Check:

   More epochs can improve model performance and AUC [OK]

Hint: More training epochs improve AUC [OK]

Common Mistakes:

• Thinking threshold tuning affects AUC
• Reducing data size harms performance
• Switching to accuracy doesn't improve AUC