hex.tree.xgboost.predict

Class XGBoostJavaBigScoreChunkPredict

java.lang.Object

hex.tree.xgboost.predict.XGBoostJavaBigScoreChunkPredict

All Implemented Interfaces:

hex.Model.BigScoreChunkPredict, XGBoostPredict, java.lang.AutoCloseable

public class XGBoostJavaBigScoreChunkPredict
extends java.lang.Object
implements XGBoostPredict, hex.Model.BigScoreChunkPredict

Nested Class Summary

Nested classes/interfaces inherited from interface hex.tree.xgboost.predict.XGBoostPredict

XGBoostPredict.OutputType

Constructor Summary

Constructors

Constructor and Description
XGBoostJavaBigScoreChunkPredict(hex.DataInfo di, XGBoostOutput output, XGBoostModel.XGBoostParameters parms, double threshold, biz.k11i.xgboost.Predictor predictor, water.fvec.Frame data)

Method Summary

Modifier and Type	Method and Description
void	close()
float[][]	predict(water.fvec.Chunk[] cs)
double[]	score0(water.fvec.Chunk[] chks, double offset, int row_in_chunk, double[] tmp, double[] preds)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

XGBoostJavaBigScoreChunkPredict

public XGBoostJavaBigScoreChunkPredict(hex.DataInfo di,
                                       XGBoostOutput output,
                                       XGBoostModel.XGBoostParameters parms,
                                       double threshold,
                                       biz.k11i.xgboost.Predictor predictor,
                                       water.fvec.Frame data)

Method Detail

score0

public double[] score0(water.fvec.Chunk[] chks,
                       double offset,
                       int row_in_chunk,
                       double[] tmp,
                       double[] preds)

Specified by:

score0 in interface hex.Model.BigScoreChunkPredict

predict

public float[][] predict(water.fvec.Chunk[] cs)

Specified by:

predict in interface XGBoostPredict

close

public void close()

Specified by:

close in interface hex.Model.BigScoreChunkPredict

Specified by:

close in interface java.lang.AutoCloseable