Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Split up SHAP from RegTree. (#8612)

Browse Source 
* Split up SHAP from `RegTree`.

Simplify the tree interface.
This commit is contained in:
Jiaming Yuan
2023-01-04 18:17:47 +08:00
committed by GitHub
parent d308124910
commit beefd28471
7 changed files with 225 additions and 220 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/predictor/cpu_predictor.cc
View File

@@ -15,6 +15,7 @@
#include "../data/gradient_index.h"
#include "../data/proxy_dmatrix.h"
#include "../gbm/gbtree_model.h"
#include "cpu_treeshap.h" // CalculateContributions
#include "predict_fn.h"
#include "xgboost/base.h"
#include "xgboost/data.h"
@@ -530,9 +531,8 @@ class CPUPredictor : public Predictor {
continue;
}
if (!approximate) {
model.trees[j]->CalculateContributions(
feats, tree_mean_values, &this_tree_contribs[0], condition,
condition_feature);
CalculateContributions(*model.trees[j], feats, tree_mean_values,
&this_tree_contribs[0], condition, condition_feature);
} else {
model.trees[j]->CalculateContributionsApprox(
feats, tree_mean_values, &this_tree_contribs[0]);