public class DeepWaterTask extends FrameTask<DeepWaterTask>
FrameTask.ExtractDenseRow
Modifier and Type | Method and Description |
---|---|
protected void |
closeLocal()
After all maps are done on a node, this is called to store the per-node model into DKV (for elastic averaging)
Otherwise, do nothing.
|
void |
map(water.fvec.Chunk[] chunks,
water.fvec.NewChunk[] outputs)
Extracts the values, applies regularization to numerics, adds appropriate offsets to categoricals,
and adapts response according to the CaseMode/CaseValue if set.
|
DeepWaterModelInfo |
model_info()
Accessor to the object containing the (final) state of the Deep Learning model
Should only be queried after calling this.doAll(Frame training)
|
protected void |
postGlobal()
After all reduces are done, the driver node calls this method to clean up
This is only needed if we're not inside a DeepWaterTask2 (which will do the reduction between replicated data workers).
|
void |
reduce(DeepWaterTask other)
Average the per-node models (for elastic averaging, already wrote them to DKV in postLocal())
This is a no-op between F/J worker threads (operate on the same weights/biases)
|
protected void |
setupLocal()
Transfer ownership from global (shared) model to local model which will be worked on
|
chunkDone, chunkInit, dinfo, getMiniBatchSize, processMiniBatch, processRow, processRow, processRow
appendables, asyncExecOnAllNodes, block, 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, modifiesVolatileVecs, onCompletion, onExceptionalCompletion, outputFrame, outputFrame, outputFrame, profile, profString, self
copyOver, getDException, hasException, logVerbose, onAck, onAckAck, setException
asBytes, clone, compute, compute1, currThrPriority, frozenType, icer, priority, read, readJSON, reloadFromBytes, write, writeJSON
__tryComplete, addToPendingCount, compareAndSetPendingCount, complete, exec, getCompleter, getPendingCount, getRawResult, setCompleter, setPendingCount, setRawResult, tryComplete
adapt, adapt, adapt, cancel, compareAndSetForkJoinTaskTag, completeExceptionally, fork, 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
public final DeepWaterModelInfo model_info()
protected void setupLocal()
setupLocal
in class FrameTask<DeepWaterTask>
public void map(water.fvec.Chunk[] chunks, water.fvec.NewChunk[] outputs)
FrameTask
map
in class FrameTask<DeepWaterTask>
protected void closeLocal()
closeLocal
in class FrameTask<DeepWaterTask>
public void reduce(DeepWaterTask other)
reduce
in class water.MRTask<DeepWaterTask>
other
- Other DeepWaterTask to reduceprotected void postGlobal()
postGlobal
in class water.MRTask<DeepWaterTask>