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>

Convenient base class for steps defining a (random) grid search.

See Also:

Serialized Form

Nested Class Summary

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

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_TRAINING_WEIGHT

Fields inherited from class ai.h2o.automl.ModelingStep

_algo, _description, _id, _ignoredConstraints, _weight

Constructor Summary

Constructors

Constructor and Description
GridStep(IAlgo algo, java.lang.String id, int cost, AutoML autoML)

Method Summary

Modifier and Type	Method and Description
protected WorkAllocations.Work	getAllocatedWork()
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)
protected water.Key<hex.grid.Grid>	makeKey(java.lang.String name, boolean withCounter)
protected WorkAllocations.Work	makeWork()
protected abstract water.Job<hex.grid.Grid>	startJob()

Methods inherited from class ai.h2o.automl.ModelingStep

aml, applyPreprocessing, canRun, getPreprocessingConfig, getTrainedModels, getTrainedModelsKeys, getWorkAllocations, initTimeConstraints, isCVEnabled, onDone, setClassBalancingParams, setCommonModelBuilderParams, setCrossValidationParams, setCustomParams, setSeed, setStoppingCriteria, setWeightingParams, startModel, startSearch

Methods inherited from class water.Iced

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

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, 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

Constructor Detail

GridStep

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

Method Detail

startJob

protected abstract water.Job<hex.grid.Grid> startJob()

Specified by:

startJob in class ModelingStep<M extends hex.Model>

makeWork

protected WorkAllocations.Work makeWork()

Specified by:

makeWork in class ModelingStep<M extends hex.Model>

getAllocatedWork

protected WorkAllocations.Work getAllocatedWork()

Specified by:

getAllocatedWork in class ModelingStep<M extends hex.Model>

makeKey

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

Specified by:

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.