XGBoostClassifier
XGBoostClassifier
A model type for constructing a eXtreme Gradient Boosting Classifier, based on XGBoost.jl, and implementing the MLJ model interface.
From MLJ, the type can be imported using
XGBoostClassifier = @load XGBoostClassifier pkg=XGBoost
Do model = XGBoostClassifier()
to construct an instance with default hyper-parameters. Provide keyword arguments to override hyper-parameter defaults, as in XGBoostClassifier(test=...)
.
Univariate classification using xgboost.
Training data
In MLJ
or MLJBase
, bind an instance model
to data with
m = machine(model, X, y)
where
X
: any table of input features, either anAbstractMatrix
or Tables.jl-compatible table.y
: is anAbstractVector
Finite
target.
Train using fit!(m, rows=...)
.
Hyper-parameters
See https://xgboost.readthedocs.io/en/stable/parameter.html.