org.tribuo.classification.experiments

Class AllTrainerOptions

java.lang.Object

org.tribuo.classification.experiments.AllTrainerOptions

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Options, ClassificationOptions<Trainer<Label>>

public class AllTrainerOptions
extends Object
implements ClassificationOptions<Trainer<Label>>

Aggregates all the classification algorithms.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	AllTrainerOptions.AlgorithmType Types of algorithms supported.

Field Summary

Fields

Modifier and Type	Field and Description
AllTrainerOptions.AlgorithmType	algorithm
CARTClassificationOptions	cartOptions
ClassificationEnsembleOptions	ensemble
HashingOptions	hashingOptions
KernelSVMOptions	kernelSVMOptions
KNNClassifierOptions	knnOptions
LibLinearOptions	liblinearOptions
LibSVMOptions	libsvmOptions
LinearSGDOptions	linearSGDOptions
MultinomialNaiveBayesOptions	mnbOptions
XGBoostOptions	xgBoostOptions

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

header

Constructor Summary

Constructors

Constructor and Description
AllTrainerOptions()

Method Summary

Modifier and Type	Method and Description
Trainer<Label>	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, getOptionsDescription, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

algorithm

@Option(longName="algorithm",
        usage="Type of learner (or base learner). Defaults to SGD_LINEAR.")
public AllTrainerOptions.AlgorithmType algorithm

cartOptions

public CARTClassificationOptions cartOptions

knnOptions

public KNNClassifierOptions knnOptions

liblinearOptions

public LibLinearOptions liblinearOptions

libsvmOptions

public LibSVMOptions libsvmOptions

mnbOptions

public MultinomialNaiveBayesOptions mnbOptions

kernelSVMOptions

public KernelSVMOptions kernelSVMOptions

linearSGDOptions

public LinearSGDOptions linearSGDOptions

xgBoostOptions

public XGBoostOptions xgBoostOptions

ensemble

public ClassificationEnsembleOptions ensemble

hashingOptions

public HashingOptions hashingOptions

Constructor Detail

AllTrainerOptions

public AllTrainerOptions()

Method Detail

getTrainer

public Trainer<Label> getTrainer()

Description copied from interface: ClassificationOptions

Constructs the trainer based on the provided arguments.

Specified by:

getTrainer in interface ClassificationOptions<Trainer<Label>>

Returns:

The trainer.