用决策树计算特征重要性
您已经使用心脏病数据集构建了一个决策树分类器,用于识别有心脏病风险的患者。现在需要通过分析特征重要性来解释模型,找出预测心脏病的关键因素,从而支持更有针对性的医疗干预。
matplotlib.pyplot 已作为 plt 导入。X_train 和 y_train 已为您预加载。
本练习是课程的一部分
Python 可解释性 AI
练习说明
- 从
model中提取特征重要性。 - 针对给定的
feature_names绘制feature_importances。
交互式实操练习
通过完成这段示例代码来试试这个练习。
model = DecisionTreeClassifier(random_state=42)
model.fit(X_train, y_train)
# Derive feature importances
feature_importances = ____
feature_names = X_train.columns
# Plot the feature importances
____
plt.show()