• Home
• ai.h2o
• h2o-genmodel-ext-xgboost
• 3.36.1.5
• source code
• TreeSHAPHelper.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

biz.k11i.xgboost.tree.TreeSHAPHelper Maven / Gradle / Ivy

package biz.k11i.xgboost.tree;

import biz.k11i.xgboost.util.FVec;
import hex.genmodel.algos.tree.TreeSHAP;
import hex.genmodel.algos.tree.TreeSHAPPredictor;

public class TreeSHAPHelper {

  public static TreeSHAPPredictor makePredictor(RegTree tree) {
    RegTreeImpl treeImpl = (RegTreeImpl) tree;
    return new TreeSHAP<>(treeImpl.getNodes(), treeImpl.getStats(), 0);
  }

}