• Home
• ai.h2o
• h2o-algos
• 3.10.5.1
• source code
• DTreeScorer.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.

hex.tree.DTreeScorer Maven / Gradle / Ivy

package hex.tree;

import water.*;

public abstract class DTreeScorer> extends MRTask {
  protected final int _ncols;
  protected final int _nclass;
  protected final int _skip;
  protected final Key[][] _treeKeys;
  protected transient CompressedTree[][] _trees;
  protected SharedTree _st;

  public DTreeScorer(int ncols, int nclass, SharedTree st, Key[][] treeKeys) {
    _ncols = ncols;
    _nclass = nclass;
    _treeKeys = treeKeys;
    _st = st;
    _skip = _st.numSpecialCols();
  }
  protected int ntrees() { return _trees.length; }

  @Override protected final void setupLocal() {
    int ntrees = _treeKeys.length;
    _trees = new CompressedTree[ntrees][];
    for (int t=0; t