Skip to content

xgboost_ray-0.0.1
Compare
Choose a tag to compare
@krfricke krfricke released this 05 Jan 10:04
v0.0.1
1acfc85
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Initial version of xgboost on ray, featuring:

  • Distributed training and predict support, tested on clusters of up to 600 nodes
  • Fault tolerance: Restarting the whole run from latest checkpoint if a node fails
  • Fault tolerance: Automatic scaling down/up when nodes die/become available again
  • Data loading from various sources (CSV, Parquet, Modin dataframes, Ray MLDataset, pandas, numpy)
  • Seamless integration with Ray Tune
  • Initial Ray placement group support
Assets 2
Loading