Distributed training of XGBoost models using Scala

The example notebooks show how you can use XGBoost with MLlib:

  • The first example shows how to embed an XGBoost model into an MLlib ML pipeline.

  • The second example shows how to use MLlib cross validation to tune an XGBoost model.

XGBoost classification with ML pipeline notebook

Open notebook in new tab

XGBoost regression with cross-validation notebook

Open notebook in new tab