Package org.tribuo.common.tree

Class AbstractTrainingNode.LeafDeterminer

java.lang.Object

org.tribuo.common.tree.AbstractTrainingNode.LeafDeterminer

Enclosing class:

AbstractTrainingNode<T extends Output<T>>

public static class AbstractTrainingNode.LeafDeterminer
extends Object

Contains parameters needed to determine whether a node is a leaf.

Constructor Summary

Constructors

Constructor	Description
LeafDeterminer(int maxDepth, float minChildWeight, float scaledMinImpurityDecrease)	Constructs a leaf determiner using the supplied parameters.

Method Summary

Modifier and Type	Method	Description
int	getMaxDepth()	Gets the maximum tree depth.
float	getMinChildWeight()	Gets the minimum example weight of a child node.
float	getScaledMinImpurityDecrease()	Gets the minimum impurity decrease necessary to split a node.

Methods inherited from class java.lang.Object

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

Constructor Details

LeafDeterminer

public LeafDeterminer(int maxDepth,
 float minChildWeight,
 float scaledMinImpurityDecrease)

Constructs a leaf determiner using the supplied parameters.

Parameters:

maxDepth - The maximum tree depth.

minChildWeight - The minimum example weight of each child node.

scaledMinImpurityDecrease - The scaled minimum impurity decrease necessary to split a node.

Method Details

getMaxDepth

public int getMaxDepth()

Gets the maximum tree depth.

Returns:

The maximum tree depth.

getMinChildWeight

public float getMinChildWeight()

Gets the minimum example weight of a child node.

Returns:

The mimimum weight of a child node.

getScaledMinImpurityDecrease

public float getScaledMinImpurityDecrease()

Gets the minimum impurity decrease necessary to split a node.

Returns:

The minimum impurity decrease to split a node.