ai.h2o.automl

Class ModelingStep<M extends hex.Model>

java.lang.Object

water.Iced<ModelingStep>

ai.h2o.automl.ModelingStep<M>

All Implemented Interfaces:

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

Direct Known Subclasses:

ModelingStep.GridStep, ModelingStep.ModelStep

public abstract class ModelingStep<M extends hex.Model>
extends water.Iced<ModelingStep>

Parent class defining common properties and common logic for actual AutoML training steps.

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ModelingStep.GridStep<M extends hex.Model> Convenient base class for steps defining a (random) grid search.
static class	ModelingStep.ModelStep<M extends hex.Model> Convenient base class for single/default model steps.

Field Summary

Fields

Modifier and Type	Field and Description
protected Algo	_algo
protected java.lang.String	_description
protected java.lang.String	_id
protected boolean	_ignoreConstraints
protected int	_weight

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	ModelingStep(Algo algo, java.lang.String id, int weight, AutoML autoML)

Method Summary

Modifier and Type	Method and Description
protected AutoML	aml()
protected boolean	canRun()
protected abstract WorkAllocations.Work	getAllocatedWork()
protected hex.Model[]	getTrainedModels()
protected water.Key<hex.Model>[]	getTrainedModelsKeys()
protected WorkAllocations	getWorkAllocations()
protected boolean	isCVEnabled()
protected abstract water.Key	makeKey(java.lang.String name, boolean withCounter)
protected abstract WorkAllocations.Work	makeWork()
protected abstract water.Job	startJob()

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

_algo

protected final Algo _algo

_id

protected final java.lang.String _id

_weight

protected int _weight

_ignoreConstraints

protected boolean _ignoreConstraints

_description

protected java.lang.String _description

Constructor Detail

ModelingStep

protected ModelingStep(Algo algo,
                       java.lang.String id,
                       int weight,
                       AutoML autoML)

Method Detail

getAllocatedWork

protected abstract WorkAllocations.Work getAllocatedWork()

makeKey

protected abstract water.Key makeKey(java.lang.String name,
                                     boolean withCounter)

makeWork

protected abstract WorkAllocations.Work makeWork()

startJob

protected abstract water.Job startJob()

aml

protected AutoML aml()

canRun

protected boolean canRun()

getWorkAllocations

protected WorkAllocations getWorkAllocations()

getTrainedModels

protected hex.Model[] getTrainedModels()

getTrainedModelsKeys

protected water.Key<hex.Model>[] getTrainedModelsKeys()

isCVEnabled

protected boolean isCVEnabled()