hex.tree.gbm.ResidualsCollector 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.tree.gbm;

import java.util.Arrays;

import hex.tree.DTreeScorer;
import water.Key;
import water.fvec.Chunk;

public class ResidualsCollector extends DTreeScorer {

  public ResidualsCollector(int ncols, int nclass, Key[][] treeKeys) {
    super(ncols, nclass, treeKeys);
  }

  @Override public void map(Chunk[] chks) {
    double[] data = new double[_ncols];
    double [] preds = new double[_nclass+1];
    int ntrees = ntrees();
    Chunk cys   = chk_resp(chks);
    for( int tidx=0; tidx