Feature importance in regression in ML Python - Model Pipeline Trace

Model Pipeline - Feature importance in regression

This pipeline shows how a regression model learns to predict a number from data and how we find which features matter most for the prediction.

Data Flow - 5 Stages

1Raw data input

1000 rows x 5 columns→Load dataset with 5 features and 1 target value→1000 rows x 6 columns

Feature1=3.2, Feature2=1.5, Feature3=0.7, Feature4=4.1, Feature5=2.0, Target=10.5

↓

2Data preprocessing

1000 rows x 6 columns→Fill missing values and scale features→1000 rows x 6 columns

Feature1=0.8, Feature2=0.3, Feature3=0.1, Feature4=1.0, Feature5=0.5, Target=10.5

↓

3Train-test split

1000 rows x 6 columns→Split data into 800 training and 200 testing rows→Training: 800 rows x 6 columns, Testing: 200 rows x 6 columns

Training row example: Feature1=0.8, ..., Target=10.5

↓

4Model training

800 rows x 5 feature columns→Train linear regression model on training data→Model with 5 learned coefficients and intercept

Coefficients: Feature1=2.5, Feature2=0.0, Feature3=1.2, Feature4=0.0, Feature5=0.8; Intercept=1.0

↓

5Feature importance calculation

Model with 5 coefficients→Calculate absolute value of coefficients as importance scores→Feature importance scores for 5 features

Importance: Feature1=2.5, Feature2=0.0, Feature3=1.2, Feature4=0.0, Feature5=0.8

Training Trace - Epoch by Epoch

Loss
15.0 |*****
10.5 |****
7.8  |***
6.0  |**
5.2  |*

Epoch	Loss ↓	Accuracy ↑	Observation
1	15.0	N/A	Initial loss is high as model starts learning
2	10.5	N/A	Loss decreases as model fits data better
3	7.8	N/A	Loss continues to decrease steadily
4	6.0	N/A	Model is converging with lower loss
5	5.2	N/A	Loss stabilizes, model has learned key patterns

Prediction Trace - 4 Layers

Layer 1: Input features

Layer 2: Multiply by coefficients

Layer 3: Add intercept

Layer 4: Output prediction

Model Quiz - 3 Questions

Which feature has the highest importance in the regression model?

A Feature4

B Feature2

C Feature1

D Feature5

Key Insight

In regression, the size of the coefficients shows how much each feature influences the prediction. Features with larger absolute coefficients are more important. Watching loss decrease during training confirms the model is learning well.

Practice

1. What does feature importance tell us in a regression model?

easy

A. The accuracy of the model on test data

B. The number of layers in the model

C. Which input features most influence the model's predictions

D. The speed at which the model trains

Solution

1. Step 1: Understand feature importance meaning

   Feature importance measures how much each input feature affects the model's output.

2. Step 2: Compare options to definition

   Only Which input features most influence the model's predictions correctly describes this concept as the influence of input features on predictions.

3. Final Answer:

   Which input features most influence the model's predictions -> Option C

4. Quick Check:

   Feature importance = input influence [OK]

Hint: Feature importance ranks inputs by their effect on output [OK]

Common Mistakes:

• Confusing feature importance with model accuracy
• Thinking it shows model speed or size
• Mixing it up with number of model layers

2. Which Python library function is commonly used to get feature importance from a Random Forest regression model?

easy

A. model.score()

B. model.predict()

C. model.fit()

D. model.feature_importances_

Solution

1. Step 1: Recall Random Forest attribute for feature importance

   Random Forest regression models in scikit-learn provide feature importance via the attribute feature_importances_.

2. Step 2: Eliminate other options

   predict() makes predictions, fit() trains the model, and score() evaluates performance, none return feature importance.

3. Final Answer:

   model.feature_importances_ -> Option D

4. Quick Check:

   Feature importance attribute = feature_importances_ [OK]

Hint: Feature importance in Random Forest is model.feature_importances_ [OK]

Common Mistakes:

• Using predict() instead of feature_importances_
• Confusing fit() with importance retrieval
• Trying to get importance from score()

3. Given this code snippet, what will be the output?

from sklearn.ensemble import RandomForestRegressor
import numpy as np

X = np.array([[1, 2], [3, 4], [5, 6], [7, 8]])
y = np.array([10, 20, 30, 40])
model = RandomForestRegressor(random_state=0)
model.fit(X, y)
print(model.feature_importances_)

medium

A. [0.3 0.7]

B. [0.7 0.3]

C. [0.5 0.5]

D. [0.0 1.0]

Solution

1. Step 1: Understand the dataset and model

   The features are two columns with increasing values, but the first feature has a slightly stronger linear relationship with the target.

2. Step 2: Interpret feature importance output

   Random Forest will assign higher importance to the feature with stronger influence, which is the first feature here.

3. Final Answer:

   [0.7 0.3] -> Option B

4. Quick Check:

   Feature 1 stronger -> higher importance [OK]

Hint: Feature with stronger effect gets higher importance [OK]

Common Mistakes:

• Assuming equal importance without checking data
• Expecting exact 0.5 values
• Confusing feature importance with coefficients

4. You run this code but get an error:

from sklearn.ensemble import RandomForestRegressor
model = RandomForestRegressor()
print(model.feature_importances_)

What is the problem?

medium

A. The model must be trained with fit() before accessing feature_importances_

B. RandomForestRegressor does not have feature_importances_ attribute

C. You must import feature_importances_ separately

D. feature_importances_ is only for classification models

Solution

1. Step 1: Identify cause of error

   Accessing feature_importances_ before training causes an AttributeError because the model has no learned features yet.

2. Step 2: Correct usage

   Call fit() on training data before accessing feature_importances_.

3. Final Answer:

   The model must be trained with fit() before accessing feature_importances_ -> Option A

4. Quick Check:

   Train model first, then get feature importance [OK]

Hint: Always fit model before checking feature_importances_ [OK]

Common Mistakes:

• Trying to access feature_importances_ before fit()
• Thinking feature_importances_ needs separate import
• Believing it's only for classification

5. You have a regression dataset with 10 features. After training a Random Forest model, you find two features have very low importance scores close to zero. What is the best next step?

hard

A. Remove those low-importance features and retrain the model

B. Increase the number of trees in the forest

C. Add more features similar to the low-importance ones

D. Ignore feature importance and keep all features

Solution

1. Step 1: Interpret low feature importance meaning

   Features with near-zero importance contribute little to predictions and may add noise or complexity.

2. Step 2: Decide action based on importance

   Removing unimportant features simplifies the model and can improve performance and interpretability.

3. Final Answer:

   Remove those low-importance features and retrain the model -> Option A

4. Quick Check:

   Drop low-importance features to simplify model [OK]

Hint: Drop features with near-zero importance to improve model [OK]

Common Mistakes:

• Adding more similar low-importance features
• Ignoring importance and keeping all features
• Only increasing trees without feature selection