hex.tree

Class DTree

java.lang.Object

water.Iced

hex.tree.DTree

All Implemented Interfaces:

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

public class DTree
extends water.Iced

A Decision Tree, laid over a Frame of Vecs, and built distributed.

This class defines an explicit Tree structure, as a collection of DTree Nodes. The Nodes are numbered with a unique _nid. Users need to maintain their own mapping from their data to a _nid, where the obvious technique is to have a Vec of _nids (ints), one per each element of the data Vecs.

Each Node has a DHistogram, describing summary data about the rows. The DHistogram requires a pass over the data to be filled in, and we expect to fill in all rows for Nodes at the same depth at the same time. i.e., a single pass over the data will fill in all leaf Nodes' DHistograms at once.

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DTree.DecidedNode
static class	DTree.LeafNode
static class	DTree.Node
static class	DTree.Split
static class	DTree.UndecidedNode

Field Summary

Fields

Modifier and Type	Field and Description
water.AutoBuffer	_abAux
boolean	_checkConstraintConsistency
int[]	_cols
int	_depth
int	_leaves
int	_len
int	_mtrys
int	_mtrys_per_tree
SharedTreeModel.SharedTreeParameters	_parms
java.util.Random	_rand
static int	NO_PARENT

Constructor Summary

Constructors

Constructor and Description
DTree(DTree tree) Copy constructor
DTree(water.fvec.Frame fr, int ncols, int mtrys, int mtrys_per_tree, long seed, SharedTreeModel.SharedTreeParameters parms)

Method Summary

Modifier and Type	Method and Description
int	actual_mtries()
CompressedTree	compress(int tid, int cls, java.lang.String[][] domains)
DTree.DecidedNode	decided(int i)
static boolean	isRootNode(DTree.Node n)
int	len()
DTree.Node	node(int i)
DTree.Node	root()
DTree.UndecidedNode	undecided(int i)

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

_len

public int _len

_leaves

public int _leaves

_depth

public int _depth

_mtrys

public final int _mtrys

_mtrys_per_tree

public final int _mtrys_per_tree

_rand

public final transient java.util.Random _rand

_cols

public final transient int[] _cols

_parms

public transient SharedTreeModel.SharedTreeParameters _parms

_checkConstraintConsistency

public boolean _checkConstraintConsistency

NO_PARENT

public static final int NO_PARENT

See Also:

Constant Field Values

_abAux

public transient water.AutoBuffer _abAux

Constructor Detail

DTree

public DTree(water.fvec.Frame fr,
             int ncols,
             int mtrys,
             int mtrys_per_tree,
             long seed,
             SharedTreeModel.SharedTreeParameters parms)

DTree

public DTree(DTree tree)

Copy constructor

Parameters:

tree -

Method Detail

actual_mtries

public int actual_mtries()

root

public final DTree.Node root()

node

public final DTree.Node node(int i)

undecided

public final DTree.UndecidedNode undecided(int i)

decided

public final DTree.DecidedNode decided(int i)

len

public final int len()

isRootNode

public static boolean isRootNode(DTree.Node n)

compress

public CompressedTree compress(int tid,
                               int cls,
                               java.lang.String[][] domains)