hex.genmodel.algos.tree

Class TreeSHAP<R,N extends ai.h2o.algos.tree.INode<R>,S extends ai.h2o.algos.tree.INodeStat>

java.lang.Object

hex.genmodel.algos.tree.TreeSHAP<R,N,S>

All Implemented Interfaces:

TreeSHAPPredictor<R>, java.io.Serializable

public class TreeSHAP<R,N extends ai.h2o.algos.tree.INode<R>,S extends ai.h2o.algos.tree.INodeStat>
extends java.lang.Object
implements TreeSHAPPredictor<R>

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TreeSHAP.PathPointer

Nested classes/interfaces inherited from interface hex.genmodel.algos.tree.TreeSHAPPredictor

TreeSHAPPredictor.Workspace

Constructor Summary

Constructors

Constructor and Description
TreeSHAP(N[] nodes)
TreeSHAP(N[] nodes, S[] stats, int rootNodeId)

Method Summary

Modifier and Type	Method and Description
float[]	calculateContributions(R feat, float[] out_contribs)
float[]	calculateContributions(R feat, float[] out_contribs, int condition, int condition_feature, TreeSHAPPredictor.Workspace workspace)
double[]	calculateInterventionalContributions(R feat, R background, double[] out_contribs, int[] catOffsets, boolean expand)
int	getWorkspaceSize()
TreeSHAP.PathPointer	makeWorkspace()

Methods inherited from class java.lang.Object

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

Constructor Detail

TreeSHAP

public TreeSHAP(N[] nodes)

TreeSHAP

public TreeSHAP(N[] nodes,
                S[] stats,
                int rootNodeId)

Method Detail

calculateContributions

public float[] calculateContributions(R feat,
                                      float[] out_contribs)

Specified by:

calculateContributions in interface TreeSHAPPredictor<R>

calculateContributions

public float[] calculateContributions(R feat,
                                      float[] out_contribs,
                                      int condition,
                                      int condition_feature,
                                      TreeSHAPPredictor.Workspace workspace)

Specified by:

calculateContributions in interface TreeSHAPPredictor<R>

calculateInterventionalContributions

public double[] calculateInterventionalContributions(R feat,
                                                     R background,
                                                     double[] out_contribs,
                                                     int[] catOffsets,
                                                     boolean expand)

Specified by:

calculateInterventionalContributions in interface TreeSHAPPredictor<R>

makeWorkspace

public TreeSHAP.PathPointer makeWorkspace()

Specified by:

makeWorkspace in interface TreeSHAPPredictor<R>

getWorkspaceSize

public int getWorkspaceSize()

Specified by:

getWorkspaceSize in interface TreeSHAPPredictor<R>