hex.genmodel.algos.tree.TreeSHAPPredictor Maven / Gradle / Ivy

Go to download

package hex.genmodel.algos.tree;

import java.io.Serializable;

public interface TreeSHAPPredictor extends Serializable  {

  float[] calculateContributions(final R feat, float[] out_contribs);

  float[] calculateContributions(final R feat,
                                 float[] out_contribs, int condition, int condition_feature,
                                 Workspace workspace);

  double[] calculateInterventionalContributions(final R feat, final R background, double[] out_contribs, int[] catOffsets, boolean expand);
  
  Workspace makeWorkspace();

  int getWorkspaceSize();

  interface Workspace {
    int getSize();
  }
  
}