hex.tree.DTreeUtilsQ Maven / Gradle / Ivy

Go to download

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

H2O Algorithms

There is a newer version: 3.46.0.6

package hex.gbm;

import hex.gbm.DTree.TreeModel.CompressedTree;

/** Toolkit class providing various useful methods for tree models */
public class DTreeUtils {

  /**
   * Score given tree on the row of data.
   *
   * @param data row of data
   * @param preds array to hold resulting prediction
   * @param ts a tree representation (single regression tree, or multi tree)
   */
  public static void scoreTree(double data[], float preds[], CompressedTree[] ts) {
    for( int c=0; c