Machine learning -- Databricks Documentation

XGBoost is a popular machine learning library designed specifically for training decision trees and random forests. For information about installing XGBoost on Databricks Runtime, or installing a custom version on Databricks Runtime ML, see these instructions. You can train XGBoost models on an individual machine or in a distributed fashion.