Package org.tribuo.common.xgboost

Provides abstract classes for interfacing with XGBoost abstracting away all the Output dependent parts.

See: Description

Interface Summary

Interface	Description
XGBoostOutputConverter<T extends Output<T>>	Converts the output of XGBoost into the appropriate prediction type.

Class Summary

Class	Description
XGBoostExternalModel<T extends Output<T>>	A Model which wraps around a XGBoost.Booster which was trained by a system other than Tribuo.
XGBoostFeatureImportance	Generate and collate feature importance information from the XGBoost model.
XGBoostFeatureImportance.XGBoostFeatureImportanceInstance	An instance of feature importance values for a single feature.
XGBoostModel<T extends Output<T>>	A Model which wraps around a XGBoost.Booster.
XGBoostTrainer<T extends Output<T>>	A Trainer which wraps the XGBoost training procedure.
XGBoostTrainer.DMatrixTuple<T extends Output<T>>	Tuple of a DMatrix, the number of valid features in each example, and the examples themselves.
XGBoostTrainer.XGBoostTrainerProvenance	Deprecated

Enum Summary

Enum	Description
XGBoostTrainer.BoosterType	The type of XGBoost model.
XGBoostTrainer.LoggingVerbosity	The logging verbosity of the native library.
XGBoostTrainer.TreeMethod	The tree building algorithm.

Package org.tribuo.common.xgboost Description

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.