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.

Sign up for GitHub

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

Jump to bottom

Rework xgboost jvm packages for spark without example #23

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

Rework xgboost jvm packages for spark without example #23

wants to merge 1 commit into from
+6,071 −8,692

Conversation

wbo4958
Copy link
Owner

@wbo4958 wbo4958 commented Jul 23, 2024

  • 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
Rework xgboost jvm packages for spark
e8ecea6 
- 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.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
1 participant
@wbo4958