Evaluation metrics (RMSE, precision@k) in ML Python - Model Pipeline Trace

Model Pipeline - Evaluation metrics (RMSE, precision@k)

This pipeline shows how we measure how well a model predicts numbers using RMSE and how well it ranks items using precision@k.

Data Flow - 5 Stages

1Data in

1000 rows x 5 columns→Collect raw data with features and true labels→1000 rows x 5 columns

Features: user age, item category; Label: true rating (3.5) or relevant item (yes/no)

↓

2Preprocessing

1000 rows x 5 columns→Clean data and split into features and labels→1000 rows x 4 columns (features), 1000 rows x 1 column (labels)

Features: [25, electronics, ...], Label: 4.0

↓

3Model Trains

800 rows x 4 columns (features), 800 rows x 1 column (labels)→Train model to predict ratings or rank items→Trained model

Model learns to predict rating 4.0 or rank items for user

↓

4Model Predicts

200 rows x 4 columns (features)→Model predicts ratings or ranks items→200 rows x 1 column (predicted ratings) or 200 rows x k items (ranked list)

Predicted rating: 3.8, Ranked items: [item5, item2, item9]

↓

5Evaluation Metrics

200 rows x 1 column (predicted ratings), 200 rows x 1 column (true ratings)→Calculate RMSE for rating predictions and precision@k for ranked items→Single RMSE value, precision@k value

RMSE: 0.45, precision@5: 0.8

Training Trace - Epoch by Epoch

Loss
1.2 |*       
0.9 | **     
0.7 |  ***   
0.6 |   ***  
0.55|    *** 
    +--------
     1 2 3 4 5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.2	N/A	Initial loss is high because model is just starting
2	0.9	N/A	Loss decreases as model learns patterns
3	0.7	N/A	Loss continues to decrease, model improving
4	0.6	N/A	Loss stabilizes, model converging
5	0.55	N/A	Small improvements, training nearing end

Prediction Trace - 5 Layers

Layer 1: Input features

Layer 2: Model prediction for rating

Layer 3: Calculate RMSE

Layer 4: Model prediction for ranking

Layer 5: Calculate precision@3

Model Quiz - 3 Questions

What does a lower RMSE value mean?

A Model is overfitting

B Predictions are more spread out

C Predictions are closer to true values

D Model is ignoring input features

Key Insight

RMSE helps us understand how close our predicted numbers are to the true numbers by measuring average error size. Precision@k tells us how good the model is at picking the right top items, which is important in recommendations.

Practice

1. What does RMSE (Root Mean Squared Error) measure in a machine learning model?

easy

A. The average size of prediction errors for continuous values

B. The percentage of correct classifications

C. The number of true positives in classification

D. The speed of the model training process

Solution

1. Step 1: Understand RMSE definition

   RMSE calculates the square root of the average squared differences between predicted and actual values, showing error size.

2. Step 2: Identify what RMSE measures

   It measures how far predictions are from actual continuous values on average, not classification accuracy or speed.

3. Final Answer:

   The average size of prediction errors for continuous values -> Option A

4. Quick Check:

   RMSE = average error size [OK]

Hint: RMSE = error size for numbers, not classification [OK]

Common Mistakes:

• Confusing RMSE with accuracy
• Thinking RMSE measures classification counts
• Assuming RMSE measures speed

2. Which of the following is the correct formula to calculate RMSE given predictions \(\hat{y}_i\) and true values \(y_i\) for \(n\) samples?

easy

A. \(\frac{\sum_{i=1}^n (y_i - \hat{y}_i)}{n}\)

B. \(\frac{1}{n} \sum_{i=1}^n |y_i - \hat{y}_i|\)

C. \(\frac{\text{True Positives}}{\text{True Positives} + \text{False Positives}}\)

D. \(\sqrt{\frac{1}{n} \sum_{i=1}^n (y_i - \hat{y}_i)^2}\)

Solution

1. Step 1: Recall RMSE formula

   RMSE is the square root of the average squared differences between true and predicted values.

2. Step 2: Match formula options

   \(\sqrt{\frac{1}{n} \sum_{i=1}^n (y_i - \hat{y}_i)^2}\) matches the RMSE formula exactly; others represent MAE, mean error, or precision.

3. Final Answer:

   \(\sqrt{\frac{1}{n} \sum_{i=1}^n (y_i - \hat{y}_i)^2}\) -> Option D

4. Quick Check:

   RMSE = sqrt(mean squared error) [OK]

Hint: RMSE = sqrt of average squared errors [OK]

Common Mistakes:

• Confusing RMSE with MAE (mean absolute error)
• Forgetting to square errors before averaging
• Mixing precision formula with RMSE

3. Given the true labels \([1, 0, 1, 1, 0]\) and model predicted scores \([0.9, 0.2, 0.8, 0.4, 0.3]\), what is the precision@3?

medium

A. 1.0

B. 0.67

C. 0.33

D. 0.5

Solution

1. Step 1: Identify top 3 predictions by score

   Top 3 scores are 0.9 (index 0), 0.8 (index 2), and 0.4 (index 3).

2. Step 2: Count true positives in top 3

   True labels at these indices are 1 (index 0), 1 (index 2), 1 (index 3). Total true positives = 3.

3. Step 3: Calculate precision@3

   Precision@3 = true positives / 3 = 3/3 = 1.0.

4. Final Answer:

   1.0 -> Option A

5. Quick Check:

   Precision@3 = 3/3 = 1.0 [OK]

Hint: Count true positives in top k predictions, divide by k [OK]

Common Mistakes:

• Counting all positives, not just top k
• Mixing precision@k with recall
• Using predicted labels instead of scores for ranking

4. You have a ranking model and calculated precision@5 as 0.4, but you expected 0.6. Which of the following could fix this issue?

medium

A. Increase the number of items in the dataset

B. Improve the model to rank more relevant items in the top 5

C. Decrease the number of true positives in the top 5

D. Use RMSE instead of precision@5

Solution

1. Step 1: Understand precision@5 meaning

   Precision@5 measures how many relevant items appear in the top 5 predictions.

2. Step 2: Identify how to increase precision@5

   To increase precision@5, the model must rank more relevant items higher within the top 5.

3. Step 3: Evaluate options

   Improve the model to rank more relevant items in the top 5 improves ranking of relevant items; A changes dataset size but not ranking; C reduces true positives, lowering precision; D changes metric but not model quality.

4. Final Answer:

   Improve the model to rank more relevant items in the top 5 -> Option B

5. Quick Check:

   Better ranking = higher precision@5 [OK]

Hint: Boost relevant items in top k to raise precision@k [OK]

Common Mistakes:

• Changing dataset size instead of model quality
• Reducing true positives mistakenly
• Confusing RMSE with precision@k

5. A regression model has predictions \([2.5, 0.0, 2.1, 7.8]\) and true values \([3.0, -0.5, 2.0, 7.5]\). Calculate the RMSE and decide if the model's average error is less than 0.5.

hard

A. RMSE = 0.65, no the error is not less than 0.5

B. RMSE = 0.55, no the error is not less than 0.5

C. RMSE = 0.39, yes the error is less than 0.5

D. RMSE = 0.35, yes the error is less than 0.5

Solution

1. Step 1: Calculate squared errors for each prediction

   (3.0-2.5)^2=0.25, (-0.5-0.0)^2=0.25, (2.0-2.1)^2=0.01, (7.5-7.8)^2=0.09. Sum=0.60.

2. Step 2: Compute mean squared error and RMSE

   Mean squared error = 0.60/4 = 0.15; RMSE = \(\sqrt{0.15}\) \(\approx\) 0.39.

3. Step 3: Compare RMSE with 0.5

   RMSE \(\approx\) 0.39 < 0.5, so average error is less than 0.5.

4. Final Answer:

   RMSE = 0.39, yes the error is less than 0.5 -> Option C

5. Quick Check:

   RMSE \(\approx\) 0.39 < 0.5 [OK]

Hint: Square errors, average, then sqrt for RMSE [OK]

Common Mistakes:

• Forgetting to square errors
• Incorrect averaging of squared errors
• Mixing RMSE with MAE