Feature importance explanation in ML Python - ML Experiment: Train & Evaluate

Experiment - Feature importance explanation

Problem:You have trained a decision tree model to predict if a person will buy a product based on features like age, income, and browsing time.

Current Metrics:Training accuracy: 90%, Validation accuracy: 85%

Issue:You want to understand which features the model thinks are most important to make predictions.

Your Task

Explain the importance of each feature used by the decision tree model and visualize it clearly.

Use the trained decision tree model only.

Do not retrain the model or change its parameters.

Hint 1

Hint 2

Hint 3

Solution

ML Python

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
import matplotlib.pyplot as plt
import numpy as np

# Sample data: Iris dataset (for demonstration)
data = load_iris()
X = data.data
y = data.target
feature_names = data.feature_names

# Train a decision tree model
model = DecisionTreeClassifier(random_state=42)
model.fit(X, y)

# Get feature importances
importances = model.feature_importances_

# Plot feature importances
indices = np.argsort(importances)[::-1]
plt.figure(figsize=(8,5))
plt.title('Feature Importances')
plt.bar(range(len(importances)), importances[indices], color='skyblue')
plt.xticks(range(len(importances)), [feature_names[i] for i in indices], rotation=45, ha='right')
plt.ylabel('Importance')
plt.tight_layout()
plt.show()

# Print feature importance values
for i in indices:
    print(f"Feature: {feature_names[i]}, Importance: {importances[i]:.3f}")

Used the trained decision tree model's feature_importances_ attribute to get importance scores.

Sorted features by importance and plotted a bar chart for easy visualization.

Printed feature names with their importance values for clear explanation.

Results Interpretation

Before: The model was a black box with no clear idea which features mattered most.

After: We have a clear list and visual chart showing which features the model uses most to decide.

Feature importance helps us understand what drives the model's decisions. It builds trust and guides us to focus on the most useful data.

Bonus Experiment

Try using permutation feature importance to explain feature importance on the same model.

💡 Hint

Use sklearn.inspection.permutation_importance to measure how shuffling each feature affects model accuracy.

Practice

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

easy

A. The time taken to train the model

B. Which features most affect the model's predictions

C. The accuracy of the model on test data

D. The total number of features in the dataset

Solution

1. Step 1: Understand the meaning of feature importance

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

2. Step 2: Compare options with definition

   Only Which features most affect the model's predictions correctly states that feature importance shows which features most affect predictions.

3. Final Answer:

   Which features most affect the model's predictions -> Option B

4. Quick Check:

   Feature importance = Most influential features [OK]

Hint: Feature importance ranks features by prediction impact [OK]

Common Mistakes:

• Confusing feature count with importance
• Mixing accuracy with feature importance
• Thinking it measures training time

2. Which of the following is the correct way to get feature importance from a trained Random Forest model in Python using scikit-learn?

easy

A. model.feature_importances_

B. model.get_feature_importance()

C. model.feature_importance_

D. model.importance_features

Solution

1. Step 1: Recall scikit-learn Random Forest attribute

   The correct attribute to access feature importance is feature_importances_ with an 's' and underscore at the end.

2. Step 2: Check each option

   model.feature_importances_ matches the exact attribute name. Others are incorrect or do not exist.

3. Final Answer:

   model.feature_importances_ -> Option A

4. Quick Check:

   Random Forest uses feature_importances_ [OK]

Hint: Remember the plural 'importances' with trailing underscore [OK]

Common Mistakes:

• Using singular 'feature_importance_'
• Calling a non-existent method
• Mixing attribute names

3. You have a dataset with features: Age, Income, and Number of Purchases. After training a Random Forest, the feature importances are: Age=0.1, Income=0.7, Number of Purchases=0.2. What does this tell you about your model?

easy

A. Age is the most important feature influencing predictions

B. All features are equally important

C. Number of Purchases has no effect on predictions

D. Income is the most important feature influencing predictions

Solution

1. Step 1: Interpret feature importance values

   Higher values mean stronger influence on model predictions. Income has 0.7, which is highest.

2. Step 2: Compare other features

   Age (0.1) and Number of Purchases (0.2) have lower importance, so less influence.

3. Final Answer:

   Income is the most important feature influencing predictions -> Option D

4. Quick Check:

   Highest importance = most influential feature [OK]

Hint: Highest importance value shows key feature [OK]

Common Mistakes:

• Thinking low values mean no effect
• Assuming equal importance despite values
• Confusing feature names

4. Given this code snippet:

from sklearn.ensemble import RandomForestClassifier
X = [[1, 2], [3, 4], [5, 6], [7, 8]]
y = [0, 1, 0, 1]
model = RandomForestClassifier(random_state=0)
model.fit(X, y)
importances = model.feature_importances_
print(importances)

What is the expected output?

medium

A. [0.5 0.5]

B. [0.7 0.3]

C. [0.0 1.0]

D. An error because data is too small

Solution

1. Step 1: Understand the dataset and model

   The dataset has two features with balanced classes. RandomForestClassifier will split importance roughly evenly.

2. Step 2: Predict feature importance values

   With simple data and no strong bias, feature importance will be close to equal, around 0.5 each.

3. Final Answer:

   [0.5 0.5] -> Option A

4. Quick Check:

   Balanced features = equal importance [OK]

Hint: Small balanced data leads to roughly equal importances [OK]

Common Mistakes:

• Expecting exact 0 or 1 values
• Assuming error due to small data
• Thinking importance sums to more than 1

5. You run this code but get an AttributeError:

from sklearn.ensemble import RandomForestClassifier
model = RandomForestClassifier()
model.feature_importances_

What is the likely cause?

medium

A. You need to import a different library for feature importance

B. RandomForestClassifier does not have feature_importances_ attribute

C. The model is not trained yet before accessing feature_importances_

D. The attribute name is misspelled

Solution

1. Step 1: Check model usage

   The model is created but not trained with fit() before accessing feature_importances_.

2. Step 2: Understand attribute availability

   Feature importance is only available after training. Accessing it before fit causes AttributeError.

3. Final Answer:

   The model is not trained yet before accessing feature_importances_ -> Option C

4. Quick Check:

   Train model before feature_importances_ [OK]

Hint: Always call fit() before feature_importances_ [OK]

Common Mistakes:

• Assuming attribute doesn't exist
• Thinking import is missing
• Misspelling attribute name