hex.tree

Class SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask

java.lang.Object

jsr166y.ForkJoinTask<java.lang.Void>

jsr166y.CountedCompleter

water.H2O.H2OCountedCompleter<T>

water.DTask<T>

water.MRTask<T>

hex.ContributionsWithBackgroundFrameTask<SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask>

hex.tree.SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask

All Implemented Interfaces:

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

Direct Known Subclasses:

DRFModel.ScoreContributionsWithBackgroundTaskDRF

Enclosing class:

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

public class SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask
extends ContributionsWithBackgroundFrameTask<SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask>

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 int[]	_catOffsets
protected boolean	_expand
protected SharedTreeModel	_model
protected water.Key<SharedTreeModel>	_modelKey
protected SharedTreeModel.SharedTreeOutput	_output
protected boolean	_outputSpace
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
ScoreContributionsWithBackgroundTask(water.Key<water.fvec.Frame> frKey, water.Key<water.fvec.Frame> backgroundFrameKey, boolean perReference, SharedTreeModel model, boolean expand, int[] catOffsets, boolean outputSpace)

Method Summary

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

_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

_expand

protected boolean _expand

_outputSpace

protected boolean _outputSpace

_catOffsets

protected int[] _catOffsets

Constructor Detail

ScoreContributionsWithBackgroundTask

public ScoreContributionsWithBackgroundTask(water.Key<water.fvec.Frame> frKey,
                                            water.Key<water.fvec.Frame> backgroundFrameKey,
                                            boolean perReference,
                                            SharedTreeModel model,
                                            boolean expand,
                                            int[] catOffsets,
                                            boolean outputSpace)

Method Detail

setupLocal

protected void setupLocal()

Overrides:

setupLocal in class water.MRTask<SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask>

fillInput

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

map

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

Specified by:

map in class ContributionsWithBackgroundFrameTask<SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask>

doModelSpecificComputation

protected void doModelSpecificComputation(double[] contribs)

addContribToNewChunk

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