Skip to content
This repository has been archived by the owner on May 6, 2024. It is now read-only.

Latest commit

 

History

History
11 lines (6 loc) · 846 Bytes

README.md

File metadata and controls

11 lines (6 loc) · 846 Bytes

spark-xgboost

Build

Spark users can use XGBoost for classification and regression tasks in a distributed environment through the excellent XGBoost4J-Spark library. As of July 2020, this integration only exposes a Scala API. A PR is open on the main XGBoost repository to add a Python equivalent, but this is still in draft.

This repository contains the Python wrapper components from that PR. By building and installing the appropriate .whl (see 'releases' in this repository), PySpark users can directly use this wrapper with the XGBoost library.

Tested with XGBoost 1.5.2

See the notebook in /examples for a practical illustration of usage.