hex.utils.MSETsk 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.utils;

import water.MRTask;
import water.fvec.Chunk;

/**
 * Created by tomasnykodym on 9/9/14.
 */
public class MSETsk extends MRTask {
  public double _resDev;
  public long   _nobs;
  public void map(Chunk prediction, Chunk response){
    for(int i = 0; i < prediction._len; ++i){
      if(prediction.isNA0(i) || response.isNA0(i))
        continue;
      double diff = prediction.at0(i) - response.at0(i);
      _resDev += diff*diff;
      ++_nobs;
    }
  }

  @Override
  public void reduce(MSETsk t){
    _resDev += t._resDev;
    _nobs += t._nobs;
  }

}