hex.genmodel.PredictContributions Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of h2o-genmodel Show documentation

H2O GenModel

There is a newer version: 3.46.0.5

package hex.genmodel;

import hex.genmodel.attributes.parameters.FeatureContribution;

import java.io.Serializable;

public interface PredictContributions extends Serializable {

  /**
   * Calculate contributions (SHAP values) for a given input row.
   * @param input input data
   * @return per-feature contributions, last value is the model bias
   */
  float[] calculateContributions(double[] input);

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

  String[] getContributionNames();

}