hex.tree.xgboost.predict

Class PredictTreeSHAPWithBackgroundTask

java.lang.Object

jsr166y.ForkJoinTask<java.lang.Void>

jsr166y.CountedCompleter

water.H2O.H2OCountedCompleter<T>

water.DTask<T>

water.MRTask<T>

hex.ContributionsWithBackgroundFrameTask<PredictTreeSHAPWithBackgroundTask>

hex.tree.xgboost.predict.PredictTreeSHAPWithBackgroundTask

All Implemented Interfaces:

java.io.Serializable, java.lang.Cloneable, java.util.concurrent.Future<java.lang.Void>, jsr166y.ForkJoinPool.ManagedBlocker, water.Freezable<PredictTreeSHAPWithBackgroundTask>

public class PredictTreeSHAPWithBackgroundTask
extends hex.ContributionsWithBackgroundFrameTask<PredictTreeSHAPWithBackgroundTask>

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class water.MRTask

water.MRTask.PostMapAction<T extends water.MRTask.PostMapAction<T>>

Nested classes/interfaces inherited from class water.DTask

water.DTask.DKeyTask<T extends water.DTask.DKeyTask,V extends water.Keyed>, water.DTask.RemoveCall

Field Summary

Fields

Modifier and Type	Field and Description
protected hex.DataInfo	_di
protected hex.Distribution	_distribution
protected XGBoostModelInfo	_modelInfo
protected hex.genmodel.algos.xgboost.XGBoostJavaMojoModel	_mojo
protected XGBoostOutput	_output
protected boolean	_outputAggregated
protected boolean	_outputSpace

Fields inherited from class water.MRTask

_appendables, _fr, _fs, _hi, _keys, _left, _lo, _nhi, _nleft, _nlo, _nrite, _rite, _run_local, _topLocal

Fields inherited from class water.DTask

_ex

Constructor Summary

Constructors

Constructor and Description
PredictTreeSHAPWithBackgroundTask(hex.DataInfo di, XGBoostModelInfo modelInfo, XGBoostOutput output, hex.Model.Contributions.ContributionsOptions options, water.fvec.Frame frame, water.fvec.Frame backgroundFrame, boolean perReference, boolean outputSpace)

Method Summary

Modifier and Type	Method and Description
protected void	addContribToNewChunk(double[] contribs, water.fvec.NewChunk[] nc)
protected void	fillInput(water.fvec.Chunk[] chks, int row, double[] input)
protected void	map(water.fvec.Chunk[] cs, water.fvec.Chunk[] bgCs, water.fvec.NewChunk[] ncs)
protected void	setupLocal()

Methods inherited from class hex.ContributionsWithBackgroundFrameTask

enoughMinMemory, estimatePerNodeMinimalMemory, estimateRequiredMemory, loadFrames, map, minMemoryPerNode, runAndGetOutput, totalFreeMemory

Methods inherited from class water.MRTask

appendables, asyncExecOnAllNodes, block, closeLocal, compute2, dfork, dfork, dfork, dfork, dfork, dinvoke, doAll, doAll, doAll, doAll, doAll, doAll, doAll, doAll, doAll, doAll, doAll, doAll, doAllNodes, getResult, getResult, isReleasable, map, map, map, map, map, map, map, map, map, map, map, modifiesVolatileVecs, onCompletion, onExceptionalCompletion, outputFrame, outputFrame, outputFrame, postGlobal, profile, profString, reduce, self, withPostMapAction

Methods inherited from class water.DTask

copyOver, getDException, hasException, logVerbose, onAck, onAckAck, setException

Methods inherited from class water.H2O.H2OCountedCompleter

asBytes, clone, compute, compute1, currThrPriority, frozenType, icer, priority, read, readJSON, reloadFromBytes, write, writeJSON

Methods inherited from class jsr166y.CountedCompleter

__tryComplete, addToPendingCount, compareAndSetPendingCount, complete, exec, getCompleter, getPendingCount, getRawResult, setCompleter, setPendingCount, setRawResult, tryComplete

Methods inherited from class jsr166y.ForkJoinTask

adapt, adapt, adapt, cancel, compareAndSetForkJoinTaskTag, completeExceptionally, fork, get, get, get, getException, getForkJoinTaskTag, getPool, getQueuedTaskCount, getSurplusQueuedTaskCount, helpQuiesce, inForkJoinPool, invoke, invokeAll, invokeAll, invokeAll, isCancelled, isCompletedAbnormally, isCompletedNormally, isDone, join, peekNextLocalTask, pollNextLocalTask, pollTask, quietlyComplete, quietlyInvoke, quietlyJoin, reinitialize, setForkJoinTaskTag, tryUnfork

Methods inherited from class java.lang.Object

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

Field Detail

_di

protected final hex.DataInfo _di

_modelInfo

protected final XGBoostModelInfo _modelInfo

_output

protected final XGBoostOutput _output

_outputAggregated

protected final boolean _outputAggregated

_outputSpace

protected final boolean _outputSpace

_distribution

protected final hex.Distribution _distribution

_mojo

protected transient hex.genmodel.algos.xgboost.XGBoostJavaMojoModel _mojo

Constructor Detail

PredictTreeSHAPWithBackgroundTask

public PredictTreeSHAPWithBackgroundTask(hex.DataInfo di,
                                         XGBoostModelInfo modelInfo,
                                         XGBoostOutput output,
                                         hex.Model.Contributions.ContributionsOptions options,
                                         water.fvec.Frame frame,
                                         water.fvec.Frame backgroundFrame,
                                         boolean perReference,
                                         boolean outputSpace)

Method Detail

setupLocal

protected void setupLocal()

Overrides:

setupLocal in class water.MRTask<PredictTreeSHAPWithBackgroundTask>

fillInput

protected void fillInput(water.fvec.Chunk[] chks,
                         int row,
                         double[] input)

addContribToNewChunk

protected void addContribToNewChunk(double[] contribs,
                                    water.fvec.NewChunk[] nc)

map

protected void map(water.fvec.Chunk[] cs,
                   water.fvec.Chunk[] bgCs,
                   water.fvec.NewChunk[] ncs)

Specified by:

map in class hex.ContributionsWithBackgroundFrameTask<PredictTreeSHAPWithBackgroundTask>