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>

Convenient base class for single/default model steps.

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_MODEL_TRAINING_WEIGHT

Fields inherited from class ai.h2o.automl.ModelingStep

_algo, _description, _id, _ignoredConstraints, _weight

Constructor Summary

Constructors

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

Method Summary

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

public static final int DEFAULT_MODEL_TRAINING_WEIGHT

See Also:

Constant Field Values

Constructor Detail

ModelStep

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

Method Detail

startJob

protected abstract water.Job<M> 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<M> makeKey(java.lang.String name,
                               boolean withCounter)

Specified by:

makeKey in class ModelingStep<M extends hex.Model>

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.