org.tribuo.common.tree

Class ExtraTreesTrainer<T extends Output<T>>

java.lang.Object

org.tribuo.ensemble.BaggingTrainer<T>

org.tribuo.common.tree.ExtraTreesTrainer<T>

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<TrainerProvenance>, Trainer<T>

public class ExtraTreesTrainer<T extends Output<T>>
extends BaggingTrainer<T>

A trainer which produces an Extremely Randomized Tree Ensemble.

Extremely Randomized Trees are similar to Random Forests, but they add an extra element of randomness in that the split points for features are also chosen randomly. As with Random Forests, feature subsampling is available at each of the nodes. An exception will be thrown if the inner trainer is not a decision tree trainer or if random splitting is turned off.

See:

 P. Geurts, D. Ernst, L. Wehenkel.
 "Extremely Randomized Trees"
 March 2006. PDF
 

Field Summary

Fields inherited from class org.tribuo.ensemble.BaggingTrainer

combiner, innerTrainer, numMembers, rng, seed, trainInvocationCounter

Fields inherited from interface org.tribuo.Trainer

DEFAULT_SEED

Constructor Summary

Constructors

Constructor and Description
ExtraTreesTrainer(DecisionTreeTrainer<T> trainer, EnsembleCombiner<T> combiner, int numMembers) Constructs an ExtraTreesTrainer with the default seed Trainer.DEFAULT_SEED.
ExtraTreesTrainer(DecisionTreeTrainer<T> trainer, EnsembleCombiner<T> combiner, int numMembers, long seed) Constructs an ExtraTreesTrainer with the supplied seed, trainer, combining function and number of members.

Method Summary

Modifier and Type	Method and Description
protected String	ensembleName()
void	postConfig() Used by the OLCUT configuration system, and should not be called by external code.
String	toString()

Methods inherited from class org.tribuo.ensemble.BaggingTrainer

getInvocationCount, getProvenance, train, trainSingleModel

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.tribuo.Trainer

train

Constructor Detail

ExtraTreesTrainer

public ExtraTreesTrainer(DecisionTreeTrainer<T> trainer,
                         EnsembleCombiner<T> combiner,
                         int numMembers)

Constructs an ExtraTreesTrainer with the default seed Trainer.DEFAULT_SEED.

Throws PropertyException if the trainer is not set to use random split points.

Parameters:

trainer - The tree trainer.

combiner - The combining function for the ensemble.

numMembers - The number of ensemble members to train.

ExtraTreesTrainer

public ExtraTreesTrainer(DecisionTreeTrainer<T> trainer,
                         EnsembleCombiner<T> combiner,
                         int numMembers,
                         long seed)

Constructs an ExtraTreesTrainer with the supplied seed, trainer, combining function and number of members.

Throws PropertyException if the trainer is not set to use random split points.

Parameters:

trainer - The tree trainer.

combiner - The combining function for the ensemble.

numMembers - The number of ensemble members to train.

seed - The RNG seed.

Method Detail

postConfig

public void postConfig()

Description copied from class: BaggingTrainer

Used by the OLCUT configuration system, and should not be called by external code.

Specified by:

postConfig in interface com.oracle.labs.mlrg.olcut.config.Configurable

Overrides:

postConfig in class BaggingTrainer<T extends Output<T>>

ensembleName

protected String ensembleName()

Overrides:

ensembleName in class BaggingTrainer<T extends Output<T>>

toString

public String toString()

Overrides:

toString in class BaggingTrainer<T extends Output<T>>