hex.genmodel

Interface PredictContributions

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

ContributionsPredictor, SharedTreeMojoModelWithContributions.SharedTreeContributionsPredictor

public interface PredictContributions
extends java.io.Serializable

Method Summary

Modifier and Type	Method and Description
float[]	calculateContributions(double[] input) Calculate contributions (SHAP values) for a given input row.
FeatureContribution[]	calculateContributions(double[] input, int topN, int bottomN, boolean compareAbs)
java.lang.String[]	getContributionNames()

Method Detail

calculateContributions

float[] calculateContributions(double[] input)

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

Parameters:

input - input data

Returns:

per-feature contributions, last value is the model bias

calculateContributions

FeatureContribution[] calculateContributions(double[] input,
                                             int topN,
                                             int bottomN,
                                             boolean compareAbs)

getContributionNames

java.lang.String[] getContributionNames()