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
int	cartMaxDepth
float	cartMinChildWeight
float	cartMinImpurityDecrease
boolean	cartPrintTree
boolean	cartRandomSplit
long	cartSeed
float	cartSplitFraction
CARTClassificationOptions.TreeType	cartTreeAlgorithm

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

cartMinChildWeight

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

cartMinImpurityDecrease

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

cartSplitFraction

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

cartRandomSplit

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

cartImpurity

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

cartPrintTree

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

cartTreeAlgorithm

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

cartSeed

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

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.