开始使用免费开始使用

用决策树计算特征重要性

您已经使用心脏病数据集构建了一个决策树分类器,用于识别有心脏病风险的患者。现在需要通过分析特征重要性来解释模型,找出预测心脏病的关键因素,从而支持更有针对性的医疗干预。

matplotlib.pyplot 已作为 plt 导入。X_trainy_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()
编辑并运行代码