hex.tree.DTreeScorer Maven / Gradle / Ivy

Go to download
package hex.tree;

import water.*;
import water.fvec.Chunk;

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

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

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