hex.genmodel.algos.tree

Class SharedTreeMojoModelWithContributions.ContributionsPredictor

java.lang.Object

hex.genmodel.algos.tree.SharedTreeMojoModelWithContributions.ContributionsPredictor

All Implemented Interfaces:

PredictContributions, java.io.Serializable

Enclosing class:

SharedTreeMojoModelWithContributions

protected static class SharedTreeMojoModelWithContributions.ContributionsPredictor
extends java.lang.Object
implements PredictContributions

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ContributionsPredictor(SharedTreeMojoModel model, TreeSHAPPredictor<double[]> treeSHAPPredictor)

Method Summary

Modifier and Type	Method and Description
float[]	calculateContributions(double[] input) Calculate contributions (SHAP values) for a given input row.
float[]	getContribs(float[] contribs)

Methods inherited from class java.lang.Object

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

Constructor Detail

ContributionsPredictor

public ContributionsPredictor(SharedTreeMojoModel model,
                              TreeSHAPPredictor<double[]> treeSHAPPredictor)

Method Detail

calculateContributions

public final float[] calculateContributions(double[] input)

Description copied from interface: PredictContributions

Calculate contributions (SHAP values) for a given input row.

Specified by:

calculateContributions in interface PredictContributions

Parameters:

input - input data

Returns:

per-feature contributions, last value is the model bias

getContribs

public float[] getContribs(float[] contribs)