org.tribuo.regression.rtree

Class CARTJointRegressionTrainer

java.lang.Object

org.tribuo.common.tree.AbstractCARTTrainer<Regressor>

org.tribuo.regression.rtree.CARTJointRegressionTrainer

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<TrainerProvenance>, DecisionTreeTrainer<Regressor>, SparseTrainer<Regressor>, Trainer<Regressor>, WeightedExamples

public class CARTJointRegressionTrainer
extends AbstractCARTTrainer<Regressor>

A Trainer that uses an approximation of the CART algorithm to build a decision tree.

Builds a single tree for all the regression dimensions.

See:

 J. Friedman, T. Hastie, & R. Tibshirani.
 "The Elements of Statistical Learning"
 Springer 2001. PDF
 

Nested Class Summary

Nested classes/interfaces inherited from class org.tribuo.common.tree.AbstractCARTTrainer

AbstractCARTTrainer.AbstractCARTTrainerProvenance

Field Summary

Fields inherited from class org.tribuo.common.tree.AbstractCARTTrainer

fractionFeaturesInSplit, maxDepth, MIN_EXAMPLES, minChildWeight, rng, seed, trainInvocationCounter

Fields inherited from interface org.tribuo.Trainer

DEFAULT_SEED

Constructor Summary

Constructors

Constructor and Description
CARTJointRegressionTrainer() Creates a CART Trainer.
CARTJointRegressionTrainer(int maxDepth) Creates a CART Trainer.
CARTJointRegressionTrainer(int maxDepth, boolean normalize) Creates a CART Trainer.
CARTJointRegressionTrainer(int maxDepth, float minChildWeight, float fractionFeaturesInSplit, RegressorImpurity impurity, boolean normalize, long seed) Creates a CART Trainer.

Method Summary

Modifier and Type	Method and Description
TrainerProvenance	getProvenance()
protected AbstractTrainingNode<Regressor>	mkTrainingNode(Dataset<Regressor> examples)
String	toString()

Methods inherited from class org.tribuo.common.tree.AbstractCARTTrainer

getFractionFeaturesInSplit, getInvocationCount, postConfig, train, train

Methods inherited from class java.lang.Object

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

Constructor Detail

CARTJointRegressionTrainer

public CARTJointRegressionTrainer(int maxDepth,
                                  float minChildWeight,
                                  float fractionFeaturesInSplit,
                                  RegressorImpurity impurity,
                                  boolean normalize,
                                  long seed)

Creates a CART Trainer.

Parameters:

maxDepth - maxDepth The maximum depth of the tree.

minChildWeight - minChildWeight The minimum node weight to consider it for a split.

fractionFeaturesInSplit - fractionFeaturesInSplit The fraction of features available in each split.

impurity - impurity The impurity function to use to determine split quality.

normalize - Normalize the leaves so each output sums to one.

seed - The seed to use for the RNG.

CARTJointRegressionTrainer

public CARTJointRegressionTrainer()

Creates a CART Trainer. Sets the impurity to the MeanSquaredError and does not normalize the outputs.

CARTJointRegressionTrainer

public CARTJointRegressionTrainer(int maxDepth)

Creates a CART Trainer. Sets the impurity to the MeanSquaredError and does not normalize the outputs.

Parameters:

maxDepth - The maximum depth of the tree.

CARTJointRegressionTrainer

public CARTJointRegressionTrainer(int maxDepth,
                                  boolean normalize)

Creates a CART Trainer. Sets the impurity to the MeanSquaredError.

Parameters:

maxDepth - The maximum depth of the tree.

normalize - Normalises the leaves so each leaf has a distribution which sums to 1.0.

Method Detail

mkTrainingNode

protected AbstractTrainingNode<Regressor> mkTrainingNode(Dataset<Regressor> examples)

Specified by:

mkTrainingNode in class AbstractCARTTrainer<Regressor>

toString

public String toString()

Overrides:

toString in class Object

getProvenance

public TrainerProvenance getProvenance()