Regression evaluation (MSE, RMSE, R²) in ML Python - Model Pipeline Trace

Model Pipeline - Regression evaluation (MSE, RMSE, R²)

This pipeline shows how a regression model learns to predict continuous values and how we check its performance using MSE, RMSE, and R² metrics.

Data Flow - 7 Stages

1Data in

1000 rows x 3 columns→Raw dataset with features and target value→1000 rows x 3 columns

Feature1=5.1, Feature2=3.5, Target=10.2

↓

2Preprocessing

1000 rows x 3 columns→Normalize features to range 0-1→1000 rows x 3 columns

Feature1=0.51, Feature2=0.35, Target=10.2

↓

3Feature Engineering

1000 rows x 3 columns→Select features only (drop target for input)→1000 rows x 2 columns

Feature1=0.51, Feature2=0.35

↓

4Train/Test Split

1000 rows x 2 columns→Split data into 800 train rows and 200 test rows→Train: 800 rows x 2 columns, Test: 200 rows x 2 columns

Train Feature1=0.51, Feature2=0.35; Test Feature1=0.62, Feature2=0.40

↓

5Model Trains

Train: 800 rows x 2 columns→Fit linear regression model to predict target→Trained model with 2 coefficients and intercept

Coefficients: [5.0, -2.6], Intercept: 8.0

↓

6Metrics Compute

Test: 200 rows x 2 columns + true targets→Calculate MSE, RMSE, and R² on test set→Metrics values: MSE=1.25, RMSE=1.12, R²=0.85

Predicted=9.8, Actual=10.0, MSE=0.04

↓

7Prediction

Single test sample with 2 features→Model predicts continuous target value→Single predicted value

Input features: [0.6, 0.4], Predicted target: 9.96

Training Trace - Epoch by Epoch

5.0 |*****
3.2 |****
2.1 |***
1.5 |**
1.25|*

Epoch	Loss ↓	Accuracy ↑	Observation
1	5.0	N/A	Initial loss is high as model starts learning
2	3.2	N/A	Loss decreases as model fits data better
3	2.1	N/A	Loss continues to decrease steadily
4	1.5	N/A	Model is converging with lower error
5	1.25	N/A	Loss stabilizes indicating good fit

Prediction Trace - 3 Layers

Layer 1: Input features

Layer 2: Linear combination

Layer 3: Output prediction

Model Quiz - 3 Questions

What does a lower MSE value indicate about the regression model?

A The model predictions are more spread out

B The model predictions are closer to actual values

C The model is overfitting the data

D The model has higher bias

Key Insight

Regression evaluation metrics like MSE, RMSE, and R² help us understand how well the model predicts continuous values. Lower MSE and RMSE mean better predictions, while R² close to 1 means the model explains most of the variation in data.

Practice

1. What does the Mean Squared Error (MSE) measure in regression evaluation?

easy

A. The total number of data points

B. The average of the squared differences between predicted and actual values

C. The correlation between features

D. The percentage of correct predictions

Solution

1. Step 1: Understand MSE definition

   MSE calculates the average of squared differences between predicted and actual values to measure error size.

2. Step 2: Compare options to definition

   Only The average of the squared differences between predicted and actual values correctly describes MSE as average squared error; others describe unrelated concepts.

3. Final Answer:

   The average of the squared differences between predicted and actual values -> Option B

4. Quick Check:

   MSE = average squared error [OK]

Hint: MSE = average squared error between prediction and truth [OK]

Common Mistakes:

• Confusing MSE with accuracy
• Thinking MSE measures correlation
• Mixing MSE with count of data points

2. Which of the following Python code snippets correctly calculates the Root Mean Squared Error (RMSE) given true values y_true and predicted values y_pred?

easy

A. rmse = np.sum((y_true - y_pred)**2)

B. rmse = np.mean(np.abs(y_true - y_pred))

C. import numpy as np
rmse = np.sqrt(np.mean((y_true - y_pred)**2))

D. rmse = np.sqrt(np.sum(y_true - y_pred))

Solution

1. Step 1: Recall RMSE formula

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

2. Step 2: Match code to formula

   import numpy as np rmse = np.sqrt(np.mean((y_true - y_pred)**2)) correctly computes squared differences, averages them, then takes square root. Others miss steps or use wrong operations.

3. Final Answer:

   import numpy as np rmse = np.sqrt(np.mean((y_true - y_pred)**2)) -> Option C

4. Quick Check:

   RMSE = sqrt(mean squared error) [OK]

Hint: RMSE = sqrt of mean squared error, use np.sqrt(np.mean(...)) [OK]

Common Mistakes:

• Using sum instead of mean
• Taking sqrt before mean
• Using absolute difference instead of squared

3. Given the true values y_true = [3, -0.5, 2, 7] and predicted values y_pred = [2.5, 0.0, 2, 8], what is the R² score?

medium

A. 1.0

B. 0.865

C. 0.75

D. 0.948

Solution

1. Step 1: Calculate R² score formula

   R² = 1 - (sum of squared residuals / total sum of squares). Calculate residuals and totals:

2. Step 2: Compute sums

   Residuals squared: (3-2.5)^2 + (-0.5-0)^2 + (2-2)^2 + (7-8)^2 = 0.25 + 0.25 + 0 + 1 = 1.5
   Total sum squares: sum((y_true - mean(y_true))^2) with mean= (3-0.5+2+7)/4=2.875
   Totals: (3-2.875)^2 + (-0.5-2.875)^2 + (2-2.875)^2 + (7-2.875)^2 = 0.015625 + 11.390625 + 0.765625 + 17.015625 = 29.1875

3. Step 3: Calculate R²

   R² = 1 - (1.5 / 29.1875) ≈ 1 - 0.0514 = 0.9486

4. Final Answer:

   0.948 -> Option D

5. Quick Check:

   R² ≈ 0.95 means good fit [OK]

Hint: R² = 1 - (SS_res / SS_tot), calculate sums carefully [OK]

Common Mistakes:

• Using sum of residuals instead of squared residuals
• Forgetting to subtract mean in total sum squares
• Mixing up numerator and denominator

4. You have this Python code to compute MSE but it gives wrong results:

def mse(y_true, y_pred):
    return sum(y_true - y_pred)**2 / len(y_true)

What is the error?

medium

A. The subtraction is done element-wise but sum is outside, causing wrong squaring

B. The function should use absolute values instead of squares

C. The length should be squared, not the sum

D. The function is missing import statements

Solution

1. Step 1: Analyze the code operation order

   The code does sum(y_true - y_pred) first, then squares the total sum, which is incorrect for MSE.

2. Step 2: Correct MSE calculation

   MSE requires squaring each difference first, then averaging. The code should square element-wise before summing.

3. Final Answer:

   The subtraction is done element-wise but sum is outside, causing wrong squaring -> Option A

4. Quick Check:

   Square differences before sum for MSE [OK]

Hint: Square each difference before summing for MSE [OK]

Common Mistakes:

• Squaring sum instead of sum of squares
• Using absolute difference instead of squared
• Ignoring element-wise operations

5. You trained two regression models on different datasets. Model A has MSE=4 and R²=0.8, Model B has MSE=2 and R²=0.6. Which model is better and why?

hard

A. Model B is better because it has lower MSE, indicating smaller average error

B. Model A is better because it has higher R², showing better fit to data

C. Both models are equally good because MSE and R² contradict

D. Cannot decide without RMSE values

Solution

1. Step 1: Understand MSE and R² meanings

   MSE measures average squared error (lower is better). R² measures proportion of variance explained (higher is better).

2. Step 2: Compare models on error and fit

   Model B has lower MSE (2 < 4), so predictions are closer on average. Model A has higher R² but larger error, meaning it fits variance better but with bigger errors.

3. Step 3: Prioritize error for prediction quality

   Lower error (MSE) usually means better prediction accuracy, so Model B is better despite lower R².

4. Final Answer:

   Model B is better because it has lower MSE, indicating smaller average error -> Option A

5. Quick Check:

   Lower MSE means better prediction accuracy [OK]

Hint: Lower MSE beats higher R² for prediction accuracy [OK]

Common Mistakes:

• Choosing model with higher R² ignoring error size
• Thinking MSE and R² always agree
• Ignoring what metric matters for task