scala - xgboost Scala API `predictContrib` 是什么？

Question

我正在将 xgboost 与 scala API 结合使用。

我曾经将 xgboost 与 R 和 python API 一起使用，最近我发现了 pred_contribs 参数。

例如，在 python API 中，文档说:

When this is True the output will be a matrix of size (nsample, nfeats + 1) with each record indicating the feature contributions (SHAP values) for that prediction. The sum of all feature contributions is equal to the raw untransformed margin value of the prediction. Note the final column is the bias term.

据我了解，这是一种 Tree SHAP直接在xgboost中实现。

我的问题是:

predictContrib 与 Scala API 中的内容相同吗？

Answer 1

是的，两者最终执行相同的代码是完全相同的。

Python:https://github.com/dmlc/xgboost/blob/master/python-package/xgboost/core.py#L1239

斯卡拉:

1. 以 JBooster 开头
2. 那个人调用 XGBoostJNI.XGBoosterPredict

最后两者调用相同的 C API