Threshold tuning in ML Python - Model Pipeline Trace

Model Pipeline - Threshold tuning

This pipeline shows how adjusting the decision threshold of a classification model affects its predictions and performance metrics. Instead of using the default 0.5 cutoff, we tune the threshold to balance precision and recall better.

Data Flow - 6 Stages

1Raw data input

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

Feature1=0.5, Feature2=1.2, ..., Label=1

↓

2Train/test split

1000 rows x 10 columns→Split data into training (80%) and testing (20%) sets→Train: 800 rows x 10 columns, Test: 200 rows x 10 columns

Train sample: Feature1=0.3, Label=0; Test sample: Feature1=0.7, Label=1

↓

3Model training

Train: 800 rows x 10 columns→Train logistic regression model on training data→Trained model

Model learns weights for each feature

↓

4Prediction probabilities

Test: 200 rows x 10 columns→Model outputs probability scores for positive class→200 rows x 1 column (probabilities)

Sample probability: 0.72

↓

5Threshold tuning

200 rows x 1 column (probabilities)→Apply different thresholds to convert probabilities to class labels→200 rows x 1 column (predicted labels)

Threshold=0.3: predicted label=1; Threshold=0.7: predicted label=0

↓

6Metric calculation

200 rows x 1 column (predicted labels), 200 rows x 1 column (true labels)→Calculate precision, recall, and accuracy for each threshold→Metrics summary per threshold

Threshold=0.5: Precision=0.8, Recall=0.7, Accuracy=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 random weights
2	0.50	0.72	Model starts learning useful patterns
3	0.40	0.80	Loss decreases and accuracy improves
4	0.35	0.83	Model converging well
5	0.33	0.85	Training stabilizes with good accuracy

Prediction Trace - 3 Layers

Layer 1: Model prediction

Layer 2: Apply threshold 0.5

Layer 3: Apply threshold 0.7

Model Quiz - 3 Questions

What happens to the number of positive predictions if we lower the threshold from 0.7 to 0.3?

A More samples are predicted positive

B Fewer samples are predicted positive

C Number of positive predictions stays the same

D Model accuracy decreases automatically

Key Insight

Threshold tuning helps customize model predictions to fit specific needs by adjusting the cutoff point for classifying positive cases. This can improve important metrics like precision or recall depending on the problem.

Practice

1. What is the main purpose of threshold tuning in machine learning classification?

easy

A. To find the best cutoff probability to decide between classes

B. To increase the size of the training dataset

C. To reduce the number of features used in the model

D. To speed up the training process

Solution

1. Step 1: Understand threshold tuning concept

   Threshold tuning is about choosing a cutoff value for predicted probabilities to decide class labels.

2. Step 2: Identify the main goal

   The goal is to find the cutoff that best separates positive and negative classes for better decisions.

3. Final Answer:

   To find the best cutoff probability to decide between classes -> Option A

4. Quick Check:

   Threshold tuning = best cutoff choice [OK]

Hint: Threshold tuning picks the cutoff to decide yes/no [OK]

Common Mistakes:

• Confusing threshold tuning with feature selection
• Thinking threshold tuning changes training data size
• Assuming threshold tuning speeds up training

2. Which of the following is the correct way to apply a threshold of 0.7 to predicted probabilities probs in Python to get binary predictions?

easy

A. preds = (probs > 0.7).astype(int)

B. preds = probs > 0.7

C. preds = int(probs > 0.7)

D. preds = probs >= 0.7

Solution

1. Step 1: Understand threshold application

   We compare each probability to 0.7 to get True/False, then convert to 0/1 integers.

2. Step 2: Check correct syntax

   Using (probs > 0.7).astype(int) converts boolean array to integer array correctly.

3. Final Answer:

   preds = (probs > 0.7).astype(int) -> Option A

4. Quick Check:

   Threshold applied with boolean then int cast [OK]

Hint: Use boolean comparison then convert to int for binary labels [OK]

Common Mistakes:

• Forgetting to convert boolean to int
• Using int() on entire array instead of element-wise
• Using >= instead of > changes threshold logic

3. Given the following code, what will be the printed F1 score after threshold tuning?

from sklearn.metrics import f1_score
probs = [0.2, 0.8, 0.6, 0.4]
true_labels = [0, 1, 1, 0]
threshold = 0.5
preds = [1 if p > threshold else 0 for p in probs]
f1 = f1_score(true_labels, preds)
print(round(f1, 2))

medium

A. 0.80

B. 0.67

C. 1.00

D. 0.50

Solution

1. Step 1: Calculate predictions with threshold 0.5

   probs > 0.5 gives preds = [0, 1, 1, 0]

2. Step 2: Compute F1 score for preds vs true_labels

   True positives = 2, false positives = 0, false negatives = 0, so F1 = 2*TP/(2*TP+FP+FN) = 2*2/(4+0+0) = 1.0, since preds and true_labels are identical.

3. Final Answer:

   1.00 -> Option C

4. Quick Check:

   Perfect match means F1 = 1.00 [OK]

Hint: Check predicted labels carefully before scoring [OK]

Common Mistakes:

• Miscomputing predictions from threshold
• Confusing precision and recall in F1 calculation
• Rounding errors in final score

4. The following code tries to tune threshold but gives an error. What is the error?

probs = [0.1, 0.4, 0.6, 0.9]
true_labels = [0, 0, 1, 1]
thresholds = [0.3, 0.5, 0.7]
best_f1 = 0
for t in thresholds:
    preds = (probs > t)
    f1 = f1_score(true_labels, preds)
    if f1 > best_f1:
        best_f1 = f1
print(best_f1)

medium

A. Thresholds list is empty

B. Missing import of f1_score

C. preds is boolean, should be integers

D. Loop variable t is not used

Solution

1. Step 1: Check code for missing imports

   The code uses f1_score but does not import it from sklearn.metrics.

2. Step 2: Identify error cause

   Without importing f1_score, Python will raise a NameError when calling f1_score.

3. Final Answer:

   Missing import of f1_score -> Option B

4. Quick Check:

   Always import functions before use [OK]

Hint: Check if all functions are imported before use [OK]

Common Mistakes:

• Assuming boolean preds cause error (they don't)
• Ignoring missing import errors
• Thinking loop variable is unused

5. You have a model predicting probabilities for a rare disease. You want to tune the threshold to catch as many sick patients as possible but avoid too many false alarms. Which approach best balances this trade-off?

hard

A. Choose threshold maximizing recall only

B. Choose threshold minimizing accuracy

C. Choose threshold maximizing precision only

D. Choose threshold maximizing F1 score

Solution

1. Step 1: Understand the trade-off

   High recall catches more sick patients but may increase false alarms; precision reduces false alarms but may miss sick patients.

2. Step 2: Identify best metric for balance

   F1 score balances precision and recall, making it best to tune threshold for this trade-off.

3. Final Answer:

   Choose threshold maximizing F1 score -> Option D

4. Quick Check:

   F1 balances recall and precision [OK]

Hint: Use F1 score to balance recall and precision [OK]

Common Mistakes:

• Maximizing recall ignores false alarms
• Maximizing precision ignores missed cases
• Minimizing accuracy is not meaningful