org.tribuo.regression.xgboost

Class TrainTest.XGBoostOptions

java.lang.Object

org.tribuo.regression.xgboost.TrainTest.XGBoostOptions

All Implemented Interfaces:

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

Enclosing class:

TrainTest

public static class TrainTest.XGBoostOptions
extends Object
implements com.oracle.labs.mlrg.olcut.config.Options

Command line options.

Field Summary

Fields

Modifier and Type	Field and Description
float	alpha
int	depth
int	ensembleSize
float	eta
float	gamma
DataOptions	general
float	lambda
float	minWeight
int	numThreads
boolean	quiet
XGBoostRegressionTrainer.RegressionType	rType
float	subsample
float	subsampleFeatures

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

header

Constructor Summary

Constructors

Constructor and Description
XGBoostOptions()

Method Summary

Modifier and Type	Method and Description
String	getOptionsDescription()

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

Field Detail

general

public DataOptions general

rType

@Option(longName="regression-metric",
        usage="Regression type to use. Defaults to LINEAR.")
public XGBoostRegressionTrainer.RegressionType rType

ensembleSize

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

alpha

@Option(charName=97,
        longName="alpha",
        usage="L1 regularization term for weights (default 0).")
public float alpha

minWeight

@Option(longName="min-weight",
        usage="Minimum sum of instance weights needed in a leaf (default 1, range [0,inf]).")
public float minWeight

depth

@Option(charName=100,
        longName="max-depth",
        usage="Max tree depth (default 6, range (0,inf]).")
public int depth

eta

@Option(charName=101,
        longName="eta",
        usage="Step size shrinkage parameter (default 0.3, range [0,1]).")
public float eta

subsampleFeatures

@Option(longName="subsample-features",
        usage="Subsample features for each tree (default 1, range (0,1]).")
public float subsampleFeatures

gamma

@Option(charName=103,
        longName="gamma",
        usage="Minimum loss reduction to make a split (default 0, range [0,inf]).")
public float gamma

lambda

@Option(charName=108,
        longName="lambda",
        usage="L2 regularization term for weights (default 1).")
public float lambda

quiet

@Option(charName=113,
        longName="quiet",
        usage="Make the XGBoost training procedure quiet.")
public boolean quiet

subsample

@Option(longName="subsample",
        usage="Subsample size for each tree (default 1, range (0,1]).")
public float subsample

numThreads

@Option(charName=116,
        longName="num-threads",
        usage="Number of threads to use (default 4, range (1, num hw threads)).")
public int numThreads

Constructor Detail

XGBoostOptions

public XGBoostOptions()

Method Detail

getOptionsDescription

public String getOptionsDescription()

Specified by:

getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options