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

Get notebook

XGBoost regression with cross-validation notebook

Get notebook