org.tribuo.classification.xgboost

Class XGBoostOptions

java.lang.Object

org.tribuo.classification.xgboost.XGBoostOptions

All Implemented Interfaces:

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

public class XGBoostOptions
extends Object
implements ClassificationOptions<XGBoostClassificationTrainer>

CLI options for training an XGBoost classifier.

Field Summary

Fields

Modifier and Type	Field and Description
float	xbgAlpha
int	xgbEnsembleSize
float	xgbEta
float	xgbGamma
float	xgbLambda
int	xgbMaxDepth
float	xgbMinWeight
int	xgbNumThreads
boolean	xgbQuiet
float	xgbSubsample
float	xgbSubsampleFeatures

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
XGBoostClassificationTrainer	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, getOptionFields, getOptions, getOptionsDescription, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

xgbEnsembleSize

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

xbgAlpha

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

xgbMinWeight

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

xgbMaxDepth

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

xgbEta

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

xgbSubsampleFeatures

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

xgbGamma

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

xgbLambda

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

xgbQuiet

@Option(longName="xgb-quiet",
        usage="Make the XGBoost training procedure quiet.")
public boolean xgbQuiet

xgbSubsample

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

xgbNumThreads

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

Constructor Detail

XGBoostOptions

public XGBoostOptions()

Method Detail

getTrainer

public XGBoostClassificationTrainer getTrainer()

Description copied from interface: ClassificationOptions

Constructs the trainer based on the provided arguments.

Specified by:

getTrainer in interface ClassificationOptions<XGBoostClassificationTrainer>

Returns:

The trainer.