Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[jvm-packages][breaking]Rework xgboost4j jvm packages for spark #10617

Draft
wants to merge 1 commit into
base: master
Choose a base branch
from

Commits on Jul 22, 2024

  1. Rework xgboost jvm packages for spark

    - Remove xgboost4j-gpu and move its functionality to xgboost4j-spark-gpu
    - Remove any soft links in xgboost4j-spark-gpu
    - Abstract an XGBoost Estimator which handles the common functionality for all xgboost estimators.
    - Support XGBoostRanker
    - Remove any unnecessary ETL in XGBoost
    - Fix the missing value usage
    - Support uber jar for xgboost4j-spark
    - Support uber jar for xgboost4j-spark-gpu
    - Rework GPU plugin
    - More sannity test to ensure the training/transform results are same between CPU and GPU.
    wbo4958 committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    82c06da View commit details
    Browse the repository at this point in the history