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 and Description
static class	CARTClassificationOptions.ImpurityType The impurity algorithm.
static class	CARTClassificationOptions.TreeType Type of decision tree algorithm.

Field Summary

Fields

Modifier and Type	Field and 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 and Description
CARTClassificationOptions()

Method Summary

Modifier and Type	Method and 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

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

formatUsage, generateTypeDescription, getAllOptions, getEnumConstantString, getOptionFields, getOptions, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

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 Detail

CARTClassificationOptions

public CARTClassificationOptions()

Method Detail

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.