hex.deeplearning

Class DeepLearning.DeepLearningDriver

java.lang.Object

jsr166y.ForkJoinTask<java.lang.Void>

jsr166y.CountedCompleter

water.H2O.H2OCountedCompleter<hex.ModelBuilder.Driver>

hex.deeplearning.DeepLearning.DeepLearningDriver

All Implemented Interfaces:

java.io.Serializable, java.lang.Cloneable, java.util.concurrent.Future<java.lang.Void>, water.Freezable<hex.ModelBuilder.Driver>

Enclosing class:

DeepLearning

public class DeepLearning.DeepLearningDriver
extends water.H2O.H2OCountedCompleter<hex.ModelBuilder.Driver>

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DeepLearning.DeepLearningDriver()

Method Summary

Methods

Modifier and Type	Method and Description
void	buildModel() Train a Deep Learning model, assumes that all members are populated If checkpoint == null, then start training a new model, otherwise continue from a checkpoint
void	computeImpl()
DeepLearningModel	trainModel(DeepLearningModel model) Train a Deep Learning neural net model

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, onCompletion, onExceptionalCompletion, setCompleter, setPendingCount, setRawResult, tryComplete

Methods inherited from class jsr166y.ForkJoinTask

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

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DeepLearning.DeepLearningDriver

public DeepLearning.DeepLearningDriver()

Method Detail

computeImpl

public void computeImpl()

buildModel

public final void buildModel()

Train a Deep Learning model, assumes that all members are populated If checkpoint == null, then start training a new model, otherwise continue from a checkpoint

trainModel

public final DeepLearningModel trainModel(DeepLearningModel model)

Train a Deep Learning neural net model

Parameters:

model - Input model (e.g., from initModel(), or from a previous training run)

Returns:

Trained model