ai.h2o.automl

Class ModelingStep.ModelStep<M extends hex.Model>

java.lang.Object

water.Iced<ModelingStep>

ai.h2o.automl.ModelingStep<M>

ai.h2o.automl.ModelingStep.ModelStep<M>

All Implemented Interfaces:

java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.Freezable<ModelingStep>

Enclosing class:

ModelingStep<M extends hex.Model>

public abstract static class ModelingStep.ModelStep<M extends hex.Model>
extends ModelingStep<M>

Step designed to build a single/default model.

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class ai.h2o.automl.ModelingStep

ModelingStep.DynamicStep<M extends hex.Model>, ModelingStep.GridStep<M extends hex.Model>, ModelingStep.ModelStep<M extends hex.Model>, ModelingStep.SeedPolicy, ModelingStep.SelectionStep<M extends hex.Model>

Field Summary

Fields

Modifier and Type	Field and Description
static int	DEFAULT_MODEL_GROUP
static int	DEFAULT_MODEL_TRAINING_WEIGHT

Fields inherited from class ai.h2o.automl.ModelingStep

_algo, _description, _id, _ignoredConstraints, _priorityGroup, _provider, _weight, _work

Constructor Summary

Constructors

Constructor and Description
ModelStep(java.lang.String provider, IAlgo algo, java.lang.String id, AutoML autoML)
ModelStep(java.lang.String provider, IAlgo algo, java.lang.String id, int priorityGroup, int weight, AutoML autoML)

Method Summary

Modifier and Type	Method and Description
protected WorkAllocations.JobType	getJobType()
abstract hex.Model.Parameters	prepareModelParameters()
protected water.Job<M>	startJob() Starts a new Job as part of this step.
protected water.Job<M>	trainModel(water.Key<M> key, hex.Model.Parameters parms)
protected water.Job<M>	trainModel(hex.Model.Parameters parms)

Methods inherited from class ai.h2o.automl.ModelingStep

aml, applyPreprocessing, canRun, equals, getAlgo, getAllocatedWork, getGlobalId, getId, getPreprocessingConfig, getPriorityGroup, getProvider, getSubStep, getTrainedModels, getTrainedModelsKeys, getWeight, getWorkAllocations, hashCode, ignores, initTimeConstraints, isCVEnabled, isResumable, iterateSubSteps, limitModelTrainingTime, makeKey, makeWork, onDone, register, run, setClassBalancingParams, setCommonModelBuilderParams, setCrossValidationParams, setCustomParams, setDistributionParameters, setSeed, setStoppingCriteria, setWeightingParams, startModel, startSearch

Methods inherited from class water.Iced

asBytes, clone, copyOver, frozenType, read, readExternal, readJSON, reloadFromBytes, toJsonBytes, toJsonString, write, writeExternal, writeJSON

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, toString, wait, wait, wait

Field Detail

DEFAULT_MODEL_TRAINING_WEIGHT

public static final int DEFAULT_MODEL_TRAINING_WEIGHT

See Also:

Constant Field Values

DEFAULT_MODEL_GROUP

public static final int DEFAULT_MODEL_GROUP

See Also:

Constant Field Values

Constructor Detail

ModelStep

public ModelStep(java.lang.String provider,
                 IAlgo algo,
                 java.lang.String id,
                 AutoML autoML)

ModelStep

public ModelStep(java.lang.String provider,
                 IAlgo algo,
                 java.lang.String id,
                 int priorityGroup,
                 int weight,
                 AutoML autoML)

Method Detail

getJobType

protected WorkAllocations.JobType getJobType()

Specified by:

getJobType in class ModelingStep<M extends hex.Model>

prepareModelParameters

public abstract hex.Model.Parameters prepareModelParameters()

startJob

protected water.Job<M> startJob()

Description copied from class: ModelingStep

Starts a new Job as part of this step.

Specified by:

startJob in class ModelingStep<M extends hex.Model>

Returns:

the newly started job.

trainModel

protected water.Job<M> trainModel(hex.Model.Parameters parms)

trainModel

protected water.Job<M> trainModel(water.Key<M> key,
                                  hex.Model.Parameters parms)

Parameters:

key - (optional) model key.

parms - the model builder params.

Returns:

a started training model.