hex.tree.xgboost.predict

Class PredictTreeSHAPSortingTask

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

hex.tree.xgboost.predict.PredictTreeSHAPSortingTask

All Implemented Interfaces:

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

public class PredictTreeSHAPSortingTask
extends 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 inherited from class hex.tree.xgboost.predict.PredictTreeSHAPTask

_di, _modelInfo, _mojo, _output

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
PredictTreeSHAPSortingTask(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, int[] contribNamesSorted, water.fvec.NewChunk[] nc)
protected void	fillInput(water.fvec.Chunk[] chks, int row, double[] input, float[] contribs, int[] contribNameIds)
void	map(water.fvec.Chunk[] chks, water.fvec.NewChunk[] nc)

Methods inherited from class hex.tree.xgboost.predict.PredictTreeSHAPTask

addContribToNewChunk, fillInput, handleOutputFormat, 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

Constructor Detail

PredictTreeSHAPSortingTask

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

Method Detail

fillInput

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

map

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

Overrides:

map in class PredictTreeSHAPTask

addContribToNewChunk

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