org.tribuo.classification.ensemble

Class ClassificationEnsembleOptions

java.lang.Object

org.tribuo.classification.ensemble.ClassificationEnsembleOptions

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Options

public class ClassificationEnsembleOptions
extends Object
implements com.oracle.labs.mlrg.olcut.config.Options

Options for building a classification ensemble.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ClassificationEnsembleOptions.EnsembleType The type of ensemble.

Field Summary

Fields

Modifier and Type	Field and Description
int	ensembleSize
long	seed
ClassificationEnsembleOptions.EnsembleType	type

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

header

Constructor Summary

Constructors

Constructor and Description
ClassificationEnsembleOptions()

Method Summary

Modifier and Type	Method and Description
Trainer<Label>	wrapTrainer(Trainer<Label> trainer) Wraps the supplied trainer using the ensemble trainer described by these options.

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, getOptionsDescription, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

type

@Option(longName="ensemble-type",
        usage="Ensemble method, options are {ADABOOST, BAGGING, RF}.")
public ClassificationEnsembleOptions.EnsembleType type

ensembleSize

@Option(longName="ensemble-size",
        usage="Number of base learners in the ensemble.")
public int ensembleSize

seed

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

Constructor Detail

ClassificationEnsembleOptions

public ClassificationEnsembleOptions()

Method Detail

wrapTrainer

public Trainer<Label> wrapTrainer(Trainer<Label> trainer)

Wraps the supplied trainer using the ensemble trainer described by these options.

Parameters:

trainer - The trainer to wrap.

Returns:

An ensemble trainer.