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

Field Summary

Fields

Modifier and Type	Field	Description
AllTrainerOptions.AlgorithmType	algorithm	Type of learner (or base learner).
CARTClassificationOptions	cartOptions	Options for CART trainers.
ClassificationEnsembleOptions	ensemble	Options for classifier ensembles.
HashingOptions	hashingOptions	Options for hashing trainers.
KernelSVMOptions	kernelSVMOptions	Options for Kernel SVM trainers.
KNNClassifierOptions	knnOptions	Options for K-NN trainers.
LibLinearOptions	liblinearOptions	Options for LibLinear trainers.
LibSVMOptions	libsvmOptions	Options for LibSVM trainers.
LinearSGDOptions	linearSGDOptions	Options for Linear SGD trainers.
MultinomialNaiveBayesOptions	mnbOptions	Options for Multinomial Naive Bayes trainers.
XGBoostOptions	xgBoostOptions	Options for XGBoost trainers.

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

Type of learner (or base learner). Defaults to SGD_LINEAR.

cartOptions

public CARTClassificationOptions cartOptions

Options for CART trainers.

knnOptions

public KNNClassifierOptions knnOptions

Options for K-NN trainers.

liblinearOptions

public LibLinearOptions liblinearOptions

Options for LibLinear trainers.

libsvmOptions

public LibSVMOptions libsvmOptions

Options for LibSVM trainers.

mnbOptions

public MultinomialNaiveBayesOptions mnbOptions

Options for Multinomial Naive Bayes trainers.

kernelSVMOptions

public KernelSVMOptions kernelSVMOptions

Options for Kernel SVM trainers.

linearSGDOptions

public LinearSGDOptions linearSGDOptions

Options for Linear SGD trainers.

xgBoostOptions

public XGBoostOptions xgBoostOptions

Options for XGBoost trainers.

ensemble

public ClassificationEnsembleOptions ensemble

Options for classifier ensembles.

hashingOptions

public HashingOptions hashingOptions

Options for hashing trainers.

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.