hex.coxph.CPHBaseTask 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.coxph;

import hex.DataInfo;
import water.DKV;
import water.Key;
import water.MRTask;
import water.fvec.Chunk;

abstract class CPHBaseTask> extends MRTask {

  private Key _dinfoKey;

  protected transient DataInfo _dinfo;

  CPHBaseTask(DataInfo dinfo) {
    _dinfoKey = dinfo._key;
  }

  @Override
  public void map(Chunk[] cs) {
    chunkInit();
    DataInfo.Row row = _dinfo.newDenseRow();
    for (int r = 0; r < cs[0]._len; r++) {
      row = _dinfo.extractDenseRow(cs, r, row);
      if (row.isBad() || row.weight == 0)
        continue;
      processRow(row);
    }
  }

  abstract protected void processRow(DataInfo.Row row);

  @Override
  protected void setupLocal(){
    _dinfo = DKV.get(_dinfoKey).get();
  }

  protected void chunkInit() {}

}