Scala API xgboost Что такое «предикатКонтриб»?

Question

Я использую xgboost с API Scala.

Я использовал xgboost с R и python API и недавно обнаружил аргумент pred_contribs.

Например, в python API документация гласит:

Если для этого параметра установлено значение True, выходные данные будут представлять собой матрицу размера (nsample, nfeats + 1), где каждая запись указывает вклады характеристик (значения SHAP) для этого прогноза. Сумма всех вкладов признаков равна необработанному значению маржи в прогнозе. Обратите внимание, что последний столбец - это термин смещения.

Насколько я понимаю, это своего рода реализация Tree SHAP непосредственно в xgboost.

Мой вопрос:

predictContrib - это то же самое в Scala API?

Answer 1

Да, это в точности то же самое, что оба заканчивают выполнение одного и того же кода.

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

Scala:

1. Начинается с JBooster
2. Это называется XGBoostJNI.XGBoosterPredict

В конце оба вызывают один и тот же C API