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 Details

    • xgbBoosterType

      @Option(longName="xgb-booster-type", usage="Weak learning algorithm.") public XGBoostTrainer.BoosterType xgbBoosterType

    • xgbTreeMethod

      @Option(longName="xgb-tree-method", usage="Tree building algorithm.") public XGBoostTrainer.TreeMethod xgbTreeMethod

    • 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.") public float xbgAlpha

    • xgbMinWeight

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

    • xgbMaxDepth

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

    • xgbEta

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

    • xgbSubsampleFeatures

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

    • xgbGamma

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

    • xgbLambda

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

    • xgbQuiet

      @Option(longName="xgb-quiet", usage="Deprecated, use xgb-loglevel.") public boolean xgbQuiet

    • xgbLogLevel

      @Option(longName="xgb-loglevel", usage="Make the XGBoost training procedure quiet.") public XGBoostTrainer.LoggingVerbosity xgbLogLevel

    • xgbSubsample

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

    • xgbNumThreads

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

  • Constructor Details

    • XGBoostOptions

      public XGBoostOptions()

  • Method Details