biz.k11i.xgboost.tree

Class TreeSHAP

java.lang.Object

biz.k11i.xgboost.tree.TreeSHAP

public class TreeSHAP
extends java.lang.Object

Warning: Experimental (alpha-level) code - needs testing for correctness & benchmarking! NOT thread safe, internal state is mutated in order to avoid memory allocation on a per-observation level

Constructor Summary

Constructors

Constructor and Description
TreeSHAP(RegTreeImpl tree)

Method Summary

Methods

Modifier and Type	Method and Description
void	calculateContributions(FVec feat, int root_id, float[] out_contribs, int condition, int condition_feature)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TreeSHAP

public TreeSHAP(RegTreeImpl tree)

Method Detail

calculateContributions

public void calculateContributions(FVec feat,
                          int root_id,
                          float[] out_contribs,
                          int condition,
                          int condition_feature)