hex.tree.dt

Class DT

java.lang.Object

water.Iced

hex.ModelBuilder<DTModel,DTModel.DTParameters,DTModel.DTOutput>

hex.tree.dt.DT

All Implemented Interfaces:

java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.Freezable

public class DT
extends hex.ModelBuilder<DTModel,DTModel.DTParameters,DTModel.DTOutput>

Decision Tree

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class hex.ModelBuilder

hex.ModelBuilder.BuilderVisibility, hex.ModelBuilder.Driver, hex.ModelBuilder.FilterCols, hex.ModelBuilder.ModelTrainingCoordinator, hex.ModelBuilder.ValidationMessage

Field Summary

Fields

Modifier and Type	Field and Description
static double	EPSILON
static double	MIN_IMPROVEMENT

Fields inherited from class hex.ModelBuilder

_coordinator, _desc, _eventPublisher, _fold, _input_parms, _job, _messages, _nclass, _offset, _orig_projection_array, _origDomains, _origNames, _origTrain, _parms, _priorClassDist, _removedCols, _response, _result, _startUpOnceModelBuilder, _train, _treatment, _valid, _vresponse, _weights

Constructor Summary

Constructors

Constructor and Description
DT(boolean startup_once)
DT(DTModel.DTParameters parameters)

Method Summary

Modifier and Type	Method and Description
hex.ModelBuilder.BuilderVisibility	builderVisibility()
void	buildNextNode(java.util.Queue<DataFeaturesLimits> limitsQueue, int nodeIndex) Build next node from the first limits in queue.
hex.ModelCategory[]	can_build()
static DataFeaturesLimits	getInitialFeaturesLimits(water.fvec.Frame data) Compute initial features limits.
boolean	isSupervised()
void	makeLeafFromNode(int[] countsByClass, int nodeIndex) Set decision value to the node.
protected void	makeModelMetrics()
protected hex.ModelBuilder.Driver	trainModelImpl()

Methods inherited from class hex.ModelBuilder

algoName, algos, canLearnFromNAs, checkCustomMetricForEarlyStopping, checkDistributions, checkEarlyStoppingReproducibility, checkMemoryFootPrint_impl, checkMemoryFootPrint, checkResponseVariable, clearInitState, clearValidationErrors, computeCrossValidation, computePriorClassDistribution, cv_buildModels, cv_canBuildMainModelInParallel, cv_computeAndSetOptimalParameters, cv_initStoppingParameters, cv_mainModelScores, cv_makeAggregateModelMetrics, cv_scoreCVModels, cv_updateOptimalParameters, defaultKey, desiredChunks, dest, error_count, error, get, getMessagesByFieldAndSeverity, getName, getSysProperty, getToEigenVec, hasFoldCol, hasOffsetCol, hasTreatmentCol, hasWeightCol, haveMojo, havePojo, hide, ignoreBadColumns, ignoreConstColumns, ignoreInvalidColumns, ignoreStringColumns, ignoreUuidColumns, info, init_adaptFrameToTrain, init_getNClass, init, initWorkspace, isClassifier, isResponseOptional, isStopped, javaName, logMe, make, make, make, makeCVMetrics, makeCVModelBuilder, makeParameters, makePojoWriter, message, nclasses, nFoldCV, nFoldWork, nModelsInParallel, nModelsInParallel, nModelsInParallel, numSpecialCols, paramName, raiseReproducibilityWarning, rebalance, remainingTimeSecs, response, schemaDirectory, separateFeatureVecs, setMaxRuntimeSecsForMainModel, setTrain, setValid, shouldReorder, smallDataSize, stop_requested, timeout, train, trainModel, trainModel, trainModelNested, trainModelNested, trainModelOnH2ONode, valid, validateBinaryResponse, validateStoppingMetric, validationErrors, validationWarnings, vresponse, warn

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

equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

EPSILON

public static final double EPSILON

See Also:

Constant Field Values

MIN_IMPROVEMENT

public static final double MIN_IMPROVEMENT

See Also:

Constant Field Values

Constructor Detail

DT

public DT(DTModel.DTParameters parameters)

DT

public DT(boolean startup_once)

Method Detail

makeLeafFromNode

public void makeLeafFromNode(int[] countsByClass,
                             int nodeIndex)

Set decision value to the node.

Parameters:

countsByClass - counts of samples of each class

nodeIndex - node index

buildNextNode

public void buildNextNode(java.util.Queue<DataFeaturesLimits> limitsQueue,
                          int nodeIndex)

Build next node from the first limits in queue. The queue is updated with children here.

Parameters:

limitsQueue - queue with feature limits for nodes

nodeIndex - index of node in the tree array

getInitialFeaturesLimits

public static DataFeaturesLimits getInitialFeaturesLimits(water.fvec.Frame data)

Compute initial features limits.

Returns:

features limits

trainModelImpl

protected hex.ModelBuilder.Driver trainModelImpl()

Specified by:

trainModelImpl in class hex.ModelBuilder<DTModel,DTModel.DTParameters,DTModel.DTOutput>

builderVisibility

public hex.ModelBuilder.BuilderVisibility builderVisibility()

Overrides:

builderVisibility in class hex.ModelBuilder<DTModel,DTModel.DTParameters,DTModel.DTOutput>

can_build

public hex.ModelCategory[] can_build()

Specified by:

can_build in class hex.ModelBuilder<DTModel,DTModel.DTParameters,DTModel.DTOutput>

isSupervised

public boolean isSupervised()

Specified by:

isSupervised in class hex.ModelBuilder<DTModel,DTModel.DTParameters,DTModel.DTOutput>

makeModelMetrics

protected final void makeModelMetrics()