是否可以在shap
Python包的shap.dependence_plot
结果中添加回归线
玩具示例:
import xgboost
import shap
# train XGBoost model
X,y = shap.datasets.adult()
model = xgboost.XGBClassifier().fit(X, y)
# compute SHAP values
explainer = shap.TreeExplainer(model)
shap_values = explainer.shap_values(X)
# The shap dependence plot
shap.dependence_plot("Age", shap_values, X)
能用一种简单的方法画一条黄土样的回归线吗
您可以尝试:
相关问题 更多 >
编程相关推荐