Enum Class XGBoostTrainer.TreeMethod

All Implemented Interfaces:
Serializable, Comparable<XGBoostTrainer.TreeMethod>, Constable
Enclosing class:
XGBoostTrainer<T extends Output<T>>

public static enum XGBoostTrainer.TreeMethod extends Enum<XGBoostTrainer.TreeMethod>
The tree building algorithm.
  • Enum Constant Details

    • AUTO

      public static final XGBoostTrainer.TreeMethod AUTO
      XGBoost chooses between EXACT and APPROX depending on dataset size.
    • EXACT

      public static final XGBoostTrainer.TreeMethod EXACT
      Exact greedy algorithm, enumerates all split candidates.
    • APPROX

      public static final XGBoostTrainer.TreeMethod APPROX
      Approximate greedy algorithm, using a quantile sketch of the data and a gradient histogram.
    • HIST

      public static final XGBoostTrainer.TreeMethod HIST
      Faster histogram optimized approximate algorithm.
    • GPU_HIST

      public static final XGBoostTrainer.TreeMethod GPU_HIST
      GPU implementation of the HIST algorithm.

      Note: GPU computation may not be supported on all platforms, and Tribuo is not tested with XGBoost GPU support.

  • Field Details

    • paramName

      public final String paramName
      The parameter name used by the XGBoost native library.
  • Method Details

    • values

      public static XGBoostTrainer.TreeMethod[] values()
      Returns an array containing the constants of this enum class, in the order they are declared.
      an array containing the constants of this enum class, in the order they are declared
    • valueOf

      public static XGBoostTrainer.TreeMethod valueOf(String name)
      Returns the enum constant of this class with the specified name. The string must match exactly an identifier used to declare an enum constant in this class. (Extraneous whitespace characters are not permitted.)
      name - the name of the enum constant to be returned.
      the enum constant with the specified name
      IllegalArgumentException - if this enum class has no constant with the specified name
      NullPointerException - if the argument is null