Package org.tribuo.common.xgboost
package org.tribuo.common.xgboost
Provides abstract classes for interfacing with XGBoost abstracting away all the
Output
dependent parts.
Note: XGBoost requires a native library, on macOS this library requires libomp (which can be installed via homebrew), on Windows this native library must be compiled into a jar as it's not contained in the official XGBoost binary on Maven Central.
-
ClassDescriptionXGBoostExternalModel<T extends Output<T>>A
Model
which wraps around a XGBoost.Booster which was trained by a system other than Tribuo.XGBoostModel<T extends Output<T>>AModel
which wraps around a XGBoost.Booster.XGBoostOutputConverter<T extends Output<T>>Converts the output of XGBoost into the appropriate prediction type.XGBoostTrainer<T extends Output<T>>ATrainer
which wraps the XGBoost training procedure.The type of XGBoost model.XGBoostTrainer.DMatrixTuple<T extends Output<T>>Tuple of a DMatrix, the number of valid features in each example, and the examples themselves.Deprecated.