hex.tree.xgboost.predict

Class XGBoostBigScoreChunkPredict

java.lang.Object

hex.tree.xgboost.predict.XGBoostBigScoreChunkPredict

All Implemented Interfaces:

hex.Model.BigScoreChunkPredict, java.lang.AutoCloseable

public class XGBoostBigScoreChunkPredict
extends java.lang.Object
implements hex.Model.BigScoreChunkPredict

Constructor Summary

Constructors

Constructor and Description
XGBoostBigScoreChunkPredict(int nclasses, float[][] preds, double threshold)

Method Summary

Modifier and Type	Method and Description
void	close()
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

XGBoostBigScoreChunkPredict

public XGBoostBigScoreChunkPredict(int nclasses,
                                   float[][] preds,
                                   double threshold)

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

close

public void close()

Specified by:

close in interface hex.Model.BigScoreChunkPredict

Specified by:

close in interface java.lang.AutoCloseable