使用部分依赖图评估影响
基于您之前的工作,接下来请探索 "CGPA" 和 "University Rating" 如何影响录取决策。我们先前的分析显示,"CGPA" 是最重要的预测因子,而 "University Rating" 的重要性最低。借助部分依赖图,您可以观察当这些特征变化时,录取概率如何变化,从而为招生委员会提供更细致的影响解读。
X_train 和 y_train 已为您预先加载。
本练习是课程的一部分
Python 可解释性 AI
交互式实操练习
通过完成这段示例代码来试试这个练习。
from sklearn.ensemble import RandomForestRegressor
import shap
model = RandomForestRegressor(random_state=42)
model.fit(X_train, y_train)
# Generate the partial dependence plot for CGPA
shap.partial_dependence_plot(____, ____, X_train)