hex.tree.xgboost.predict

Class XGBoostNativeBigScoreChunkPredict

java.lang.Object

hex.tree.xgboost.predict.XGBoostNativeBigScoreChunkPredict

All Implemented Interfaces:

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

public class XGBoostNativeBigScoreChunkPredict
extends java.lang.Object
implements XGBoostPredictContrib, 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
XGBoostNativeBigScoreChunkPredict(XGBoostModelInfo modelInfo, XGBoostModel.XGBoostParameters parms, hex.DataInfo di, BoosterParms boosterParms, double threshold, XGBoostOutput output, water.fvec.Frame fr, water.fvec.Chunk[] chks)

Method Summary

Modifier and Type	Method and Description
void	close()
float[][]	predict(water.fvec.Chunk[] cs)
float[][]	predictContrib(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

XGBoostNativeBigScoreChunkPredict

public XGBoostNativeBigScoreChunkPredict(XGBoostModelInfo modelInfo,
                                         XGBoostModel.XGBoostParameters parms,
                                         hex.DataInfo di,
                                         BoosterParms boosterParms,
                                         double threshold,
                                         XGBoostOutput output,
                                         water.fvec.Frame fr,
                                         water.fvec.Chunk[] chks)

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

predictContrib

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

Specified by:

predictContrib in interface XGBoostPredictContrib

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