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.
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.

Package org.tribuo.common.xgboost Description

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.