Package org.tribuo.classification.dtree

Class CARTClassificationOptions

java.lang.Object

org.tribuo.classification.dtree.CARTClassificationOptions

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Options, ClassificationOptions<CARTClassificationTrainer>

public class CARTClassificationOptions
extends Object
implements ClassificationOptions<CARTClassificationTrainer>

Options for building a classification tree trainer.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	CARTClassificationOptions.ImpurityType	The impurity algorithm.
static enum	CARTClassificationOptions.TreeType	Type of decision tree algorithm.

Field Summary

Fields

Modifier and Type	Field	Description
CARTClassificationOptions.ImpurityType	cartImpurity	Impurity measure to use.
int	cartMaxDepth	Maximum depth in the decision tree.
float	cartMinChildWeight	Minimum child weight.
float	cartMinImpurityDecrease	Minimum impurity decrease.
boolean	cartPrintTree	Prints the decision tree.
boolean	cartRandomSplit	Whether to choose split points for features at random.
long	cartSeed	RNG seed.
float	cartSplitFraction	Fraction of features in split.
CARTClassificationOptions.TreeType	cartTreeAlgorithm	Tree algorithm to use (options are CART).

Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options

header

Constructor Summary

Constructors

Constructor	Description
CARTClassificationOptions()

Method Summary

Modifier and Type	Method	Description
String	getOptionsDescription()
CARTClassificationTrainer	getTrainer()	Constructs the trainer based on the provided arguments.

Methods inherited from class java.lang.Object

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

Field Details

cartMaxDepth

@Option(longName="cart-max-depth",
        usage="Maximum depth in the decision tree.")
public int cartMaxDepth

Maximum depth in the decision tree.

cartMinChildWeight

@Option(longName="cart-min-child-weight",
        usage="Minimum child weight.")
public float cartMinChildWeight

Minimum child weight.

cartMinImpurityDecrease

@Option(longName="cart-min-impurity-decrease",
        usage="Minimum impurity decrease.")
public float cartMinImpurityDecrease

Minimum impurity decrease.

cartSplitFraction

@Option(longName="cart-split-fraction",
        usage="Fraction of features in split.")
public float cartSplitFraction

Fraction of features in split.

cartRandomSplit

@Option(longName="cart-random-split",
        usage="Whether to choose split points for features at random.")
public boolean cartRandomSplit

Whether to choose split points for features at random.

cartImpurity

@Option(longName="cart-impurity",
        usage="Impurity measure to use. Defaults to GINI.")
public CARTClassificationOptions.ImpurityType cartImpurity

Impurity measure to use. Defaults to GINI.

cartPrintTree

@Option(longName="cart-print-tree",
        usage="Prints the decision tree.")
public boolean cartPrintTree

Prints the decision tree.

cartTreeAlgorithm

@Option(longName="cart-tree-algorithm",
        usage="Tree algorithm to use (options are CART).")
public CARTClassificationOptions.TreeType cartTreeAlgorithm

Tree algorithm to use (options are CART).

cartSeed

@Option(longName="cart-seed",
        usage="RNG seed.")
public long cartSeed

RNG seed.

Constructor Details

CARTClassificationOptions

public CARTClassificationOptions()

Method Details

getOptionsDescription

public String getOptionsDescription()

Specified by:

getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options

getTrainer

public CARTClassificationTrainer getTrainer()

Description copied from interface: ClassificationOptions

Constructs the trainer based on the provided arguments.

Specified by:

getTrainer in interface ClassificationOptions<CARTClassificationTrainer>

Returns:

The trainer.