Skip to content
This repository was archived by the owner on Dec 15, 2025. It is now read-only.

Add XGboost classification workload #628

Merged
carsonwang merged 5 commits into from
Jul 31, 2020
Merged

Add XGboost classification workload #628

carsonwang merged 5 commits into from
Jul 31, 2020

Conversation

@xwu-intel
Copy link

@xwu-intel xwu-intel commented Jul 15, 2020 •
edited
Loading

  • Reuse GradientBoostedTreeDataGenerator to generate dataset
  • Read dataset and convert to ml.LabeledPoint and to ml.DataFrame
  • Call XGBoost and passed in params for training
  • Call XGBoost prediction and print test error

XGBoost4J-Spark requires Apache Spark 2.4+

@gczsjdy
Copy link
Contributor

gczsjdy commented Jul 15, 2020

Could you add this in docs?

@carsonwang
Copy link
Collaborator

carsonwang commented Jul 20, 2020

why is travis failing ? @xwu99

@xwu-intel
Copy link
Author

xwu-intel commented Jul 20, 2020

why is travis failing ? @xwu99

XGBoost4J-Spark requires Apache Spark 2.4+

@bobjiang82 bobjiang82 mentioned this pull request Jul 30, 2020
Open
@carsonwang carsonwang merged commit 0362c34 into Intel-bigdata:master Jul 31, 2020
@carsonwang
Copy link
Collaborator

carsonwang commented Jul 31, 2020

Merge this one first.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@xwu-intel @gczsjdy @carsonwang