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

  • Field Details

    • general

      public DataOptions general

    • rType

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

    • ensembleSize

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

    • alpha

      @Option(charName='a', 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='d', longName="max-depth", usage="Max tree depth (default 6, range (0,inf]).") public int depth

    • eta

      @Option(charName='e', 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='g', longName="gamma", usage="Minimum loss reduction to make a split (default 0, range [0,inf]).") public float gamma

    • lambda

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

    • quiet

      @Option(charName='q', 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='t', longName="num-threads", usage="Number of threads to use (default 4, range (1, num hw threads)).") public int numThreads

  • Constructor Details

  • Method Details

    • getOptionsDescription

      public String getOptionsDescription()
      Specified by:
      getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options