ai.h2o.automl

Class ModelingStep.GridStep<M extends hex.Model>

java.lang.Object

water.Iced<ModelingStep>

ai.h2o.automl.ModelingStep<M>

ai.h2o.automl.ModelingStep.GridStep<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.GridStep<M extends hex.Model>
extends ModelingStep<M>

Step designed to build multiple models using a (random) grid search.

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_GRID_GROUP
static int	DEFAULT_GRID_TRAINING_WEIGHT
protected static int	GRID_STOPPING_ROUND_FACTOR

Fields inherited from class ai.h2o.automl.ModelingStep

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

Constructor Summary

Constructors

Constructor and Description
GridStep(java.lang.String provider, IAlgo algo, java.lang.String id, AutoML autoML)
GridStep(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()
protected water.Job<hex.grid.Grid>	hyperparameterSearch(water.Key<hex.grid.Grid> key, hex.Model.Parameters baseParms, java.util.Map<java.lang.String,java.lang.Object[]> searchParms)
protected water.Job<hex.grid.Grid>	hyperparameterSearch(hex.Model.Parameters baseParms, java.util.Map<java.lang.String,java.lang.Object[]> searchParms)
boolean	isResumable()
protected water.Key<hex.grid.Grid>	makeKey(java.lang.String name, boolean withCounter)
abstract hex.Model.Parameters	prepareModelParameters()
abstract java.util.Map<java.lang.String,java.lang.Object[]>	prepareSearchParameters()
protected void	setSearchCriteria(hex.grid.HyperSpaceSearchCriteria.RandomDiscreteValueSearchCriteria searchCriteria, hex.Model.Parameters baseParms)
protected water.Job<hex.grid.Grid>	startJob() Starts a new Job as part of this step.

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, iterateSubSteps, limitModelTrainingTime, 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_GRID_TRAINING_WEIGHT

public static final int DEFAULT_GRID_TRAINING_WEIGHT

See Also:

Constant Field Values

DEFAULT_GRID_GROUP

public static final int DEFAULT_GRID_GROUP

See Also:

Constant Field Values

GRID_STOPPING_ROUND_FACTOR

protected static final int GRID_STOPPING_ROUND_FACTOR

See Also:

Constant Field Values

Constructor Detail

GridStep

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

GridStep

public GridStep(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>

isResumable

public boolean isResumable()

Overrides:

isResumable in class ModelingStep<M extends hex.Model>

prepareModelParameters

public abstract hex.Model.Parameters prepareModelParameters()

prepareSearchParameters

public abstract java.util.Map<java.lang.String,java.lang.Object[]> prepareSearchParameters()

startJob

protected water.Job<hex.grid.Grid> 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.

makeKey

protected water.Key<hex.grid.Grid> makeKey(java.lang.String name,
                                           boolean withCounter)

Overrides:

makeKey in class ModelingStep<M extends hex.Model>

hyperparameterSearch

protected water.Job<hex.grid.Grid> hyperparameterSearch(hex.Model.Parameters baseParms,
                                                        java.util.Map<java.lang.String,java.lang.Object[]> searchParms)

hyperparameterSearch

protected water.Job<hex.grid.Grid> hyperparameterSearch(water.Key<hex.grid.Grid> key,
                                                        hex.Model.Parameters baseParms,
                                                        java.util.Map<java.lang.String,java.lang.Object[]> searchParms)

Parameters:

key - optional grid key

baseParms - ModelBuilder parameter values that are common across all models in the search.

searchParms - hyperparameter search space,

Returns:

the started hyperparameter search job.

setSearchCriteria

protected void setSearchCriteria(hex.grid.HyperSpaceSearchCriteria.RandomDiscreteValueSearchCriteria searchCriteria,
                                 hex.Model.Parameters baseParms)