hex.tree.xgboost.predict

Class PredictTreeSHAPTask

java.lang.Object

jsr166y.ForkJoinTask<java.lang.Void>

jsr166y.CountedCompleter

water.H2O.H2OCountedCompleter<T>

water.DTask<T>

water.MRTask<PredictTreeSHAPTask>

hex.tree.xgboost.predict.PredictTreeSHAPTask

All Implemented Interfaces:

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

Direct Known Subclasses:

PredictTreeSHAPSortingTask

public class PredictTreeSHAPTask
extends water.MRTask<PredictTreeSHAPTask>

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 XGBoostModelInfo	_modelInfo
protected hex.genmodel.algos.xgboost.XGBoostJavaMojoModel	_mojo
protected XGBoostOutput	_output
protected boolean	_outputAggregated

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
PredictTreeSHAPTask(hex.DataInfo di, XGBoostModelInfo modelInfo, XGBoostOutput output, hex.Model.Contributions.ContributionsOptions options)

Method Summary

Modifier and Type	Method and Description
protected void	addContribToNewChunk(float[] contribs, water.fvec.NewChunk[] nc)
protected void	fillInput(water.fvec.Chunk[] chks, int row, double[] input, float[] contribs)
protected void	handleOutputFormat(MutableOneHotEncoderFVec rowFVec, float[] contribs, float[] output)
void	map(water.fvec.Chunk[] chks, water.fvec.NewChunk[] nc)
protected void	setupLocal()

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

_mojo

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

Constructor Detail

PredictTreeSHAPTask

public PredictTreeSHAPTask(hex.DataInfo di,
                           XGBoostModelInfo modelInfo,
                           XGBoostOutput output,
                           hex.Model.Contributions.ContributionsOptions options)

Method Detail

setupLocal

protected void setupLocal()

Overrides:

setupLocal in class water.MRTask<PredictTreeSHAPTask>

fillInput

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

map

public void map(water.fvec.Chunk[] chks,
                water.fvec.NewChunk[] nc)

Overrides:

map in class water.MRTask<PredictTreeSHAPTask>

handleOutputFormat

protected void handleOutputFormat(MutableOneHotEncoderFVec rowFVec,
                                  float[] contribs,
                                  float[] output)

addContribToNewChunk

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