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	Description
static enum	AllTrainerOptions.AlgorithmType	Types of algorithms supported.

Field Summary

Fields

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

Method Summary

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

getOptionsDescription

Field Details

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 Details

AllTrainerOptions

public AllTrainerOptions()

Method Details

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.