org.tribuo.classification.sgd.kernel

Class KernelSVMOptions

java.lang.Object

org.tribuo.classification.sgd.kernel.KernelSVMOptions

All Implemented Interfaces:

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

public class KernelSVMOptions
extends Object
implements ClassificationOptions<KernelSVMTrainer>

Options for using the KernelSVMTrainer.

See:

 Shalev-Shwartz S, Singer Y, Srebro N, Cotter A
 "Pegasos: Primal Estimated Sub-Gradient Solver for SVM"
 Mathematical Programming, 2011.
 

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	KernelSVMOptions.KernelEnum The kernel types.

Field Summary

Fields

Modifier and Type	Field and Description
double	kernelDegree
int	kernelEpochs
double	kernelGamma
double	kernelIntercept
KernelSVMOptions.KernelEnum	kernelKernel
double	kernelLambda
int	kernelLoggingInterval

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

header

Constructor Summary

Constructors

Constructor and Description
KernelSVMOptions()

Method Summary

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

kernelIntercept

@Option(longName="kernel-intercept",
        usage="Intercept in kernel function. Defaults to 1.0.")
public double kernelIntercept

kernelDegree

@Option(longName="kernel-degree",
        usage="Degree in polynomial kernel function. Defaults to 1.0.")
public double kernelDegree

kernelGamma

@Option(longName="kernel-gamma",
        usage="Gamma value in kernel function. Defaults to 1.0.")
public double kernelGamma

kernelEpochs

@Option(longName="kernel-epochs",
        usage="Number of SGD epochs. Defaults to 5.")
public int kernelEpochs

kernelKernel

@Option(longName="kernel-kernel",
        usage="Kernel function. Defaults to LINEAR.")
public KernelSVMOptions.KernelEnum kernelKernel

kernelLambda

@Option(longName="kernel-lambda",
        usage="Lambda value in gradient optimisation. Defaults to 0.01.")
public double kernelLambda

kernelLoggingInterval

@Option(longName="kernel-logging-interval",
        usage="Log the objective after <int> examples. Defaults to 100.")
public int kernelLoggingInterval

Constructor Detail

KernelSVMOptions

public KernelSVMOptions()

Method Detail

getTrainer

public KernelSVMTrainer getTrainer()

Description copied from interface: ClassificationOptions

Constructs the trainer based on the provided arguments.

Specified by:

getTrainer in interface ClassificationOptions<KernelSVMTrainer>

Returns:

The trainer.