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

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, java.lang.Object workspace)
int	getWorkspaceSize()
java.lang.Object	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,
                                      java.lang.Object workspace)

Specified by:

calculateContributions in interface TreeSHAPPredictor<R>

makeWorkspace

public java.lang.Object makeWorkspace()

Specified by:

makeWorkspace in interface TreeSHAPPredictor<R>

getWorkspaceSize

public int getWorkspaceSize()

Specified by:

getWorkspaceSize in interface TreeSHAPPredictor<R>