Skip to content

v0.2.5

Choose a tag to compare

@jinlow jinlow released this 08 May 01:09
· 352 commits to main since this release
9fc288f

This release adds the method parameter to the predict_contributions metric. This allows either "average" (the method xgboost uses to calculate the contribution matrix when approx_contribs is True), or "weight" to be specified. The "average" method averages the leaf weights across all nodes, and then uses this for calculating the contributions. The "weight" method instead uses the internal leaf weights, to determine how a feature impacts the final score. Both methods result in a contribution matrix, that if summed by each row, is equivalent to the output of the predict method.