hex.genmodel.algos.tree

Class TreeSHAPEnsemble<R>

java.lang.Object

hex.genmodel.algos.tree.TreeSHAPEnsemble<R>

All Implemented Interfaces:

TreeSHAPPredictor<R>, java.io.Serializable

public class TreeSHAPEnsemble<R>
extends java.lang.Object
implements TreeSHAPPredictor<R>

See Also:

Serialized Form

Nested Class Summary

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

TreeSHAPPredictor.Workspace

Constructor Summary

Constructors

Constructor and Description
TreeSHAPEnsemble(java.util.Collection<TreeSHAPPredictor<R>> predictors, float initPred)

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()
TreeSHAPPredictor.Workspace	makeWorkspace()

Methods inherited from class java.lang.Object

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

Constructor Detail

TreeSHAPEnsemble

public TreeSHAPEnsemble(java.util.Collection<TreeSHAPPredictor<R>> predictors,
                        float initPred)

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 TreeSHAPPredictor.Workspace makeWorkspace()

Specified by:

makeWorkspace in interface TreeSHAPPredictor<R>

getWorkspaceSize

public int getWorkspaceSize()

Specified by:

getWorkspaceSize in interface TreeSHAPPredictor<R>