hex.tree

Class SharedTreeModelWithContributions.ScoreContributionsTask

java.lang.Object

jsr166y.ForkJoinTask<java.lang.Void>

jsr166y.CountedCompleter

water.H2O.H2OCountedCompleter<T>

water.DTask<T>

water.MRTask<SharedTreeModelWithContributions.ScoreContributionsTask>

hex.tree.SharedTreeModelWithContributions.ScoreContributionsTask

All Implemented Interfaces:

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

Direct Known Subclasses:

DRFModel.ScoreContributionsTaskDRF, SharedTreeModelWithContributions.ScoreContributionsSortingTask

Enclosing class:

SharedTreeModelWithContributions<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

public class SharedTreeModelWithContributions.ScoreContributionsTask
extends water.MRTask<SharedTreeModelWithContributions.ScoreContributionsTask>

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 SharedTreeModel	_model
protected water.Key<SharedTreeModel>	_modelKey
protected SharedTreeModel.SharedTreeOutput	_output
protected hex.genmodel.algos.tree.TreeSHAPPredictor<double[]>	_treeSHAP

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
ScoreContributionsTask(SharedTreeModel model)

Method Summary

Modifier and Type	Method and Description
protected void	addContribToNewChunk(float[] contribs, water.fvec.NewChunk[] nc)
protected void	doModelSpecificComputation(float[] contribs)
protected void	fillInput(water.fvec.Chunk[] chks, int row, double[] input, float[] contribs)
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

_modelKey

protected final water.Key<SharedTreeModel> _modelKey

_model

protected transient SharedTreeModel _model

_output

protected transient SharedTreeModel.SharedTreeOutput _output

_treeSHAP

protected transient hex.genmodel.algos.tree.TreeSHAPPredictor<double[]> _treeSHAP

Constructor Detail

ScoreContributionsTask

public ScoreContributionsTask(SharedTreeModel model)

Method Detail

setupLocal

protected void setupLocal()

Overrides:

setupLocal in class water.MRTask<SharedTreeModelWithContributions.ScoreContributionsTask>

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<SharedTreeModelWithContributions.ScoreContributionsTask>

doModelSpecificComputation

protected void doModelSpecificComputation(float[] contribs)

addContribToNewChunk

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