Package org.tribuo.common.xgboost
package org.tribuo.common.xgboost
Provides abstract classes for interfacing with XGBoost abstracting away all the
Output
dependent parts.
N.B.: XGBoost4J wraps the native C implementation of xgboost that links to various C libraries, including libgomp and glibc (on Linux). If you're running on Alpine, which does not natively use glibc, you'll need to install glibc into the container. On the macOS binary on Maven Central is compiled without OpenMP support, meaning that XGBoost is single threaded on macOS. You can recompile the macOS binary with OpenMP support after installing libomp from homebrew if necessary.
-
ClassDescriptionXGBoostExternalModel<T extends Output<T>>A
Model
which wraps around a XGBoost.Booster which was trained by a system other than Tribuo.Generate and collate feature importance information from the XGBoost model.An instance of feature importance values for a single feature.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.The logging verbosity of the native library.The tree building algorithm.Deprecated.